我试图在 pytorch 中堆叠一维张量,但堆栈函数似乎将它们解释为二维方阵。任何想法如何将一维张量堆叠成一个新的一维张量?再现性:a = torch.randn([2])b = torch.randn([3])c = torch.stack([a, b]) # want a (5,) tensorRuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 2 and 3 in dimension 1 at c:\new-builder_3\win-wheel\pytorch\aten\src\th\generic/THTensorMath.cpp:3616如果我将 a 和 b 解压缩为 (2,1) 和 (3,1) 张量或压缩它们(应该无效),错误仍然存在。这似乎也很奇怪,因为它们在未压缩时都应该是维度 1 中的大小 1(并且打印它们的大小将准确反映这一点),但错误仍然出现相同的字符对字符。此外,在维度 1 中堆叠只会使其说“...维度 0 中的 2 和 3...”。这一切都在 python 3.5.4,pytorch 0.4.1 中
添加回答
举报
0/150
提交
取消