使用pytorch的nonzero函数报warning

1.先放一下报的warning：

.\torch\csrc\utils\python_arg_parser.cpp:756: UserWarning: This overload of nonzero is deprecated： nonzero(Tensor input, , Tensor out)
Consider using one of the following signatures instead: nonzero(Tensor input, , bool as_tuple)
注意：这个警告是我把pytorch从1.4升级到1.5版本后才出现的。之前使用1.4版本未出现该问题。

2.问题出在哪儿

index0 = torch.nonzero(1 - t_index).reshape(1, -1)[0]
index1 = torch.nonzero(t_index).reshape(1, -1)[0]

报错里面其实也写得比较清楚，意思就是这样使用nonzero函数不对，得照着它提示的那样用。
我就？？？
我也没有按照它第一行那样用啊。但是无奈报warning，就让人很烦。

3.解决方法

那就只好照它提示的那样改了呗。在使用nonzero函数时加入as_tuple这个参数。修改后代码如下：

index0 = torch.nonzero(1 - t_index, as_tuple=False).reshape(1, -1)[0]
index1 = torch.nonzero(t_index, as_tuple =False).reshape(1, -1)[0]

其实这里as_tuple未true的时候意思就是返回一个bool元组。而我们一般都是想让其返回一个tensor，因此这里该参数就赋值为false。
至此，该warning就被消除掉了！

参考