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
网友评论