I have a linear layer which is nn.Linear(2000,50,bias=False). The weight matrix is 50x2000. I want to constrict each column of the weight matrix sum to 1. Specifically, w[0][0] + w[1][0] + … + w[49][0] = 1. How can I do this in pytorch ?