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

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

参考文献

[1]使用Pytorch的nonzero函数报warning： UserWarning: This overload of nonzero is deprecated