Expected 3-dimensional input for 3-dimensional weight [32, 1, 5], but got 2-dimensional input of size [5, 2048] instead
时间: 2024-04-28 11:25:51 浏览: 184
Rapid fabrication of terahertz lens via three-dimensional printing technology
这个错误通常是由于你尝试将一个2维的张量与一个3维的张量做矩阵乘法而导致的。根据你的错误信息,你的权重张量的形状是[32, 1, 5],这意味着它是一个3维的张量,而你输入的张量形状为[5, 2048],是一个2维的张量。因此,你需要将输入的张量的形状转换为3维的张量,以便与权重张量进行乘法运算。你可以使用unsqueeze()函数将输入的张量增加一个维度,使其形状变为[1, 5, 2048],然后再进行乘法运算。具体的代码实现可能会根据你的具体情况而有所不同,但是这个错误的原因应该是类似的。
阅读全文