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

最新推荐文章于 2023-11-13 17:43:26 发布

原创最新推荐文章于 2023-11-13 17:43:26 发布 · 1.3k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #pytorch

报错专栏专栏收录该内容

14 篇文章

订阅专栏

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

部署运行你感兴趣的模型镜像

我们在用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就被消除掉了！