Pytorch报错warning:UserWarning: This overload of nonzero is deprecated

一穷二白到年薪百万

于 2021-04-07 20:50:07 发布

阅读量1.2k

点赞数

CC 4.0 BY-SA版权

分类专栏：报错专栏文章标签： python pytorch

本文链接：https://blog.youkuaiyun.com/zfhsfdhdfajhsr/article/details/115495540

报错专栏专栏收录该内容

14 篇文章

订阅专栏

本文介绍了在使用PyTorch的nonzero函数时遇到的警告，并解释了为何需要将as_tuple参数设置为False以获取正确的tensor结果。通过修正这一细节，避免了不必要的警告，提高了代码的正确性和效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

我们在用pytorch查找矩阵中不为0元素索引时，会用到nonzero方法。但用这个方法时会报警告。其实使我们在python中用习惯了，没用注意到pytorch的语法。改正如下

m = torch.ones((3, 4))
print(m)
row = torch.nonzero(m)[:, 0]
columns = torch.nonzero(m)[:, 1]
m[row, columns] = 0
print(m)
# 以上的用法会报警告，正确使用方法如下
row = torch.nonzero(m, as_tuple=False)[:, 0]
columns = torch.nonzero(m, as_tuple=False)[:, 1]
# 为什么会这样改
这里as_tuple未true的时候意思就是返回一个bool元组。而我们一般都是想让其返回一个tensor，因此这里该参数就赋值为false。
至此，该warning就被消除掉了！