1

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

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

我努力了:

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

谢谢 :)

4

1 回答 1

1

PyTorch 张量有克隆方法,似乎可以满足您的需求:doc

仅复制第二维:

A = torch.zeros(4, 3, 40)
B = torch.clone(A[0,:,0])
于 2021-04-02T10:51:55.080 回答