标量tensor转numpy数组时在pycharm调试下显示异常「建议收藏」

2022-11-02 17:00:02 浏览数 (1)

最近发现了一个问题,在标量tensor转numpy数组之后,在pycharm调试的过程中,我想看一下这个数组的值,却发现显示异常。

代码语言:javascript复制
import numpy as np
import torch


a = torch.tensor(5)
b = a.numpy()
print(b)

如上面这个代码,在断点调试的时候,b这个数组的array显示出现异常

可能还是numpy的数组在定义显示的时候,是根据shape来的吧,而这个时候这个shape是一个空值,所以就有了这个无法显示的异常。

解决的方法也很简单,将

代码语言:javascript复制
a = torch.tensor(5)

改为

代码语言:javascript复制
a = torch.tensor(5).view(-1)

这样就可以了,但是其实本质上是把标量变成了矩阵。

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。

发布者:全栈程序员栈长,转载请注明出处:https://javaforall.cn/180308.html原文链接:https://javaforall.cn

0 人点赞