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