在运行pytorh的过程中一次产生了如题两个错误,错误来自于nn.BatchNorm2d(N),其中N=channels
然而输入必须是一个四维Tensor,在我的程序中输入Tensor是一个1*47的向量,一个batchsize=100,所以起初的输入是一个100*47的Tensor,然而pytorch是不识别这种输入的,解决办法:
将输入重新排列
input = input.view(-1, 1,1,47)
问题解决
在运行pytorh的过程中一次产生了如题两个错误,错误来自于nn.BatchNorm2d(N),其中N=channels
然而输入必须是一个四维Tensor,在我的程序中输入Tensor是一个1*47的向量,一个batchsize=100,所以起初的输入是一个100*47的Tensor,然而pytorch是不识别这种输入的,解决办法:
将输入重新排列
input = input.view(-1, 1,1,47)
问题解决