ptrblck:

Could you check the shape


Hi, @ptrblck. I have checked the shape.  The ConvNd weight.shape is defined as：

self.weight = Parameter(torch.Tensor(
out_channels, in_channels // groups, *kernel_size))

So the shape of features[40].weight.grad is the same as weight’s shape.

(Cout,Cin//groups, *kernel_size)

ModelOutputs(net, layer=40): shape: (N, Cout, Hout, Wout)
I think it explains why the shape is different. Am I right?