将值复制到空张量中

Question

我创建了一个特定大小的张量，即 C=[4, 3, 40]。我想将之前张量的值复制到这个张量中。

我想复制张量 A = [4, 3] 中的所有值（所以完全使用这个张量），并且只复制来自张量 B = [4, 40, 1] 的二维值（所以只有 40 个值）。 A 和 B 都是 FloatTensors。

我试过了：

ten_size = torch.zeros(4, 3, 40)

但这会生成一个全为零的张量（显然）。我不知道如何从张量 A 和 B 中提取特定值并将其输入到 ten_size 中，这样我就没有全零张量。

然后我尝试了

sec_dim = torch.clone(x_t[1]) 给我一个大小为 [40,1] 的张量。有没有办法可以从这些值和大小为 [4,3] 的张量 A 创建张量，以获得大小为 [4,3,40] 的张量？

这可能吗？还是需要更多信息来回答这个问题？我也在使用 pytorch

谢谢:)

Answer 1

PyTorch 张量具有似乎可以满足您的需求的 clone 方法：doc

仅复制第二维：

A = torch.zeros(4, 3, 40)
B = torch.clone(A[0,:,0])