我创建了一个特定大小的张量,即 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
谢谢 :)