only one element tensors can be converted to Python scalars

最新推荐文章于 2021-11-29 17:21:38 发布

AI算法网奇

最新推荐文章于 2021-11-29 17:21:38 发布

阅读量1.6w

点赞数

分类专栏： pytorch知识宝典

本文链接：https://blog.csdn.net/jacke121/article/details/82830040

版权

503 篇文章 228 订阅 ¥29.90 ¥99.00

订阅专栏

这篇博客讨论了在PyTorch中，仅有一个元素的张量可以使用item()方法转换为Python标量，而多元素张量则无法直接转换。文章通过示例代码说明了这种限制，并指出了尝试嵌套转换可能导致的异常情况。

摘要由CSDN通过智能技术生成

only one element tensors can be converted to Python scalars

维度为1,1的能item()

维度多的就不能转换：

loss=torch.Tensor([1,1])

loss.item()

import torch
a=[[1,2,3],[2,3,4],[1]]
b=torch.Tensor(a)
print(b)

发现以下代码就能引发异常，因为不能嵌套

import torch

b=[]
a=[[1,2,3],[2,3,4],[1,2,3]]
c=torch.tensor(a, dtype=torch.long)

b.append(c)

d=torch.tensor(b, dtype=torch.long)
print(d)

关注

专栏目录