美文网首页
torch item()方法

torch item()方法

作者: winddy_akoky | 来源:发表于2020-01-04 20:07 被阅读0次

item方法的使用:如果tensor只有一个元素那么调用item方法的时候就是将tensor转换成python的scalars;如果tensor不是单个元素的话那就会引发ValueError,如下面:


b.item()
Traceback (most recent call last):
    b.item()
ValueError: only one element tensors can be converted to Python scalars
 
torch.sum(b)
Out: tensor(276.)
torch.sum(b).item()
Out: 276.0

相关文章

网友评论

      本文标题:torch item()方法

      本文链接:https://www.haomeiwen.com/subject/zcyfactx.html