“Torch.cuda.randn” Respostas de código

Torch.cuda.randn

shape = torch.Size((300, 300))
x = torch.cuda.FloatTensor(shape)
torch.randn(shape, out=x)