Identity weights for 2d convolution

vision
2

nn.init.dirac_ would do this

3 Likes