如何将 pytorch 张量从 [a, b*c] 重塑为 [b,a,c]？

【问题标题】：How do I reshape pytorch tensor from [a, b*c] to [b,a,c]?如何将 pytorch 张量从 [a, b*c] 重塑为 [b,a,c]？
【发布时间】：2021-03-13 11:59:50
【问题描述】：

我收到此代码错误，我不确定如何重塑我的张量。

RuntimeError: Expected 3-dimensional input for 3-dimensional weight [32, 35, 2], but got 2-dimensional input of size [35, 64] instead

【问题讨论】：

【解决方案1】：

您可以使用 torch 中的reshape 函数，如下所示：

a = torch.rand((35,64))
b = torch.reshape(a, (32,35,2))

【讨论】：