Sorry, but could you give an example? Do you create an empty tensor then initialize it like below? Is it correct?
A = torch.empty(5, 7, requires_grad=True).type(torch.cuda.FloatTensor)
torch.nn.init.normal_(A)