“torch.cuda.randn” Réponses codées

torch.cuda.randn

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