如何将torch.tensor数据类型转化为python的内置类型?

在tensor后面加.item()即可。

import torch

a = torch.LongTensor([10])
b = a.data      # 还是tensor
c = a.item()    # int类型,而且只有单个元素才可以用.item()转化
print(b, type(b))
print(c, type(c))

'''输出
tensor([10]) <class 'torch.Tensor'>
10 <class 'int'>
'''
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章