Как преобразовать тензор Pytorch в массив NumPy?

Question

У меня есть тензор факела

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

Как я могу получить его в numpy?

Что-то вроде

b = a.tonumpy()

вывод должен быть таким же, как если бы я сделал

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

Answer 1

Вам могут пригодиться следующие две функции.

1. torch.Tensor.numpy ()
2. torch.from_numpy ()

Answer 2

Еще один полезный способ:

a = torch(0.1, device: cuda)

a.cpu().data.numpy()

Answer: array(0.1, dtype=float32)

Answer 3

скопировано с pytorch doc :

a = torch.ones(5)
print(a)

тензор ([1., 1., 1., 1., 1.])

b = a.numpy()
print(b)

[1.1. 1. 1. 1.]