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

Не используйте malloc в C ++-коде (если вы действительно не знаете, что делаете)

root = new Node;

и

tmp = new Node;

Проблема в том, что, поскольку вы используете malloc, конструктор для Node::map не вызван. Используя new, убедитесь, что вызываются все необходимые конструкторы.

-1
задан feedMe 22 January 2019 в 09:00
поделиться

2 ответа

скопировано из документа pytorch :

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

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

b = a.numpy()
print(b)

[1. 1. 1. 1. 1.]

0
ответ дан Gulzar 22 January 2019 в 09:00
поделиться

Вы можете найти полезными следующие две функции.

  1. torch.Tensor.numpy ()
  2. torch.from_numpy ()
0
ответ дан Wasi Ahmad 22 January 2019 в 09:00
поделиться
Другие вопросы по тегам:

Похожие вопросы: