Identity matrix weights intialization

1

How can I initialize the weights of a linear layer with identity matrix?

1 Like
2 
l = nn.Linear(3,3)
l.weight.data.copy_(torch.eye(3))
7 Likes