トーチテンソルがあります
a = torch.randn(1, 2, 3, 4, 5)
numpyで取得するにはどうすればよいですか?
何かのようなもの
b = a.tonumpy()
出力は私がしたのと同じでなければなりません
b = np.random.randn(1, 2, 3, 4, 5)
別の便利な方法:
a = torch(0.1, device: cuda) a.cpu().data.numpy() Answer: array(0.1, dtype=float32)