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