torch.detach() torch.item()
-
torch.detach()
用于将当前的tensor从计算图中取出
通俗点解释就是,经过detach返回的张量将不会进行反向传播计算梯度
-
torch.item()
用于将单个的tensor值,返回成python中的一个数据形式,多维的tensor就不行了,多维可以用torch.tolist()
x = torch.tensor([1.0])
x.item()
>>>1.0
a = torch.randn(2, 2)
a.tolist()
>>>[[0.012766935862600803, 0.5415473580360413],
[-0.08909505605697632, 0.7729271650314331]]
a[0,0].tolist()
>>>0.012766935862600803
-
字典中的.item()
以列表返回可遍历的(键, 值)元组数组
dict = {'a':'15岁',
'b':'14岁',
'c':'2岁',
}
for key,values in dict.items():
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
