Как получить значение из тензора? - PullRequest
0 голосов
/ 01 мая 2020

Моя python функция возвращает tensor([102],cuda::0) подобную структуру. Как мне получить первое значение из этого?

1 Ответ

0 голосов
/ 01 мая 2020

Нашел это на форумах pytorch :

a.data.cpu().numpy()[0]

, где a - тензор.

...