如何将一个张量的值插入另一个张量？

Question

我正在尝试将张量 y 插入到张量 x 最终维度 (y_dim) 中。最终张量的大小应为 (100, 16, 16, 1)，其中 y 的值放置在 100 x 0 维度中的每个维度中

import torch

y_dim = 1
x = torch.randn(100, 16, 16, y_dim)
#OR x = torch.randn(100, 16, 16)
y = torch.randn(100)

Xy = torch.cat((x, y), dim=3)

Answer 1

我认为您在理解张量和维度方面遗漏了一些东西。最简单的方法是将您的张量 x 视为包含 100 宽度和高度映射 16 的批处理，即 100 16x16-maps。因此，您正在操作一个包含 100*16*16 元素的张量。另一方面，您的 y 包含 100 标量值，它具有 100 元素。

我把问题转给你：

您将如何连接 100 16x16-maps 和 100 标量值？

---

以上问题没有答案。不过，有些事情可以做，可以对y 进行假设以执行连接：

• 如果你有一个张量 y 也包含 16x16 映射，那么是的，这个操作是可以实现的：

  >>> x = torch.rand(100, 16, 16)
  >>> y = torch.rand(100, 16, 16)
  
  >>> torch.cat((x, y)).shape
  torch.Size([200, 16, 16])
  

• 如果您考虑问题中的y，您可以将100 标量值扩展为16x16 映射。并且，然后与x 连接：

  >>> x = torch.rand(100, 16, 16)
  >>> y = torch.rand(100)
  
  >>> y_r = y[:,  None, None].repeat(1, 16, 16)
  >>> torch.cat((x, y_r))
  torch.Size([200, 16, 16])