torch.item()的用法
torch数据类型的item方法是得到只有一个元素张量里面的元素值。
如下:
>>> x = torch.tensor(4)
>>> x.item()
4
如果对包含多个元素的torch.tensor用item()方法,则会报错如下:
>>> x = torch.tensor([1,2,3,4])
>>> x.item()
Traceback (most recent call last):
File "<stdin>
原创
2021-03-25 18:10:52 ·
5475 阅读 ·
3 评论