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]=0print(m)#以上的用