torch item()方法
2020-01-04 本文已影响0人
winddy_akoky
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
