web-dev-qa-db-ja.com

pytorchテンソルをnumpy配列に変換する方法は?

トーチテンソルがあります

a = torch.randn(1, 2, 3, 4, 5)

numpyで取得するにはどうすればよいですか?

何かのようなもの

b = a.tonumpy()

出力は私がしたのと同じでなければなりません

b = np.random.randn(1, 2, 3, 4, 5)
2
Gulzar

別の便利な方法:

a = torch(0.1, device: cuda)

a.cpu().data.numpy()

Answer: array(0.1, dtype=float32)
1
Eyshika