How to feed a 3D tensor into Neural Network Forward() pass

3

Hi; wondering if I can do the same thing for nn.Conv2D and nn.ConvTranspose2d ?

Right now; I did x = torch.randn(batch_size, add_features, channels, in_features, in_features) but it raises error states that RuntimeError: Expected 4-dimensional input for 4-dimensional weight