Weight shape: [Cout, Cin, Kh, Kw] — same as PyTorch nn.Conv2d.
[Cout, Cin, Kh, Kw]
final Tensor weight;