1. 程式人生 > >Pytorch RuntimeError: should contain 1 elements not 64/ValueError: expected 4D input (got 2D input)

Pytorch RuntimeError: should contain 1 elements not 64/ValueError: expected 4D input (got 2D input)

在執行pytorh的過程中一次產生了如題兩個錯誤,錯誤來自於nn.BatchNorm2d(N),其中N=channels

然而輸入必須是一個四維Tensor,在我的程式中輸入Tensor是一個1*47的向量,一個batchsize=100,所以起初的輸入是一個100*47的Tensor,然而pytorch是不識別這種輸入的,解決辦法:

將輸入重新排列

input = input.view(-1, 1,1,47)

問題解決