我試圖在 pytorch 中堆疊一維張量,但堆棧函數(shù)似乎將它們解釋為二維方陣。任何想法如何將一維張量堆疊成一個(gè)新的一維張量?再現(xiàn)性: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) 張量或壓縮它們(應(yīng)該無(wú)效),錯(cuò)誤仍然存在。這似乎也很奇怪,因?yàn)樗鼈冊(cè)谖磯嚎s時(shí)都應(yīng)該是維度 1 中的大小 1(并且打印它們的大小將準(zhǔn)確反映這一點(diǎn)),但錯(cuò)誤仍然出現(xiàn)相同的字符對(duì)字符。此外,在維度 1 中堆疊只會(huì)使其說(shuō)“...維度 0 中的 2 和 3...”。這一切都在 python 3.5.4,pytorch 0.4.1 中
如何在pytorch中堆疊一維向量
ibeautiful
2021-09-25 16:19:46