item的定义、局限以及使用
- item()是可以将tensor的值转换为python的数字,但是很局限,只可以转换一个tensor的值
import torch
a=torch.randn(3,3)
a_sum=torch.sum(a)
print(a_sum)
print("*"*100)
print(a_sum.item())
出错的时候/使用错误:tensor是矩阵或者向量就会出错
a_max=torch.max(a,0)
# a_max.item()
print(a_max[1].item())
对于字典来说:item将字典转换为元组
person={'name':'lizhong','age':'26','city':'BeiJing','blog':'www.jb51.net'}
for key,value in person.items():
print('key=',key,',value=',value)