Hi there, I’m trying to feed 3D volumes through a NN which has a Conv3D layer.
For simplicity I have:

self.conv1 = nn.Conv3d(125, 2, 3)

and in the forward:

return self.conv1(x)

My volume is 125x256x256. When I try to feed a random tensor:

n=Net()
i=torch.randn(1,125,256,256)
o=n(i)

I get:

RuntimeError: Expected 5-dimensional input for 5-dimensional weight [2, 125, 3, 3, 3], but got input of size [1, 125, 256, 256] instead

I do not understand why is it asking a 5-dimensional input