Input dimension

vision
1

Can someone in this example the input has to be 4-dimensional?

28%20PM
28%20PM1088×1182 123 KB

There is an error message saying that the input has to be 4-dimensional if my input is anything but 4-dimensional…

Thanks a lot!

2

the blocked lines are:

    x = self.fc3(x)
    return x

sorry about that…

3

The Conv2d class only takes 4 dimensional inputs that’s why.

The inputs should be formated this way :
[batch_size, channels, height, width] if your want your forward to work.

For more information see the doc :slight_smile:
https://pytorch.org/docs/stable/nn.html