UserWarning: This overload of nonzero is deprecated: nonzero(Tensor input, Tensor out) 解决

问题:

../torch/csrc/utils/python_arg_parser.cpp:738: 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)

 解决：

你会发现你加上as_tuple也不行，torch.nonzero()也不行，官方推荐我们用torch.where函数代替，如何代替见例子.

>>> import torch
>>> array=torch.Tensor([0,1,1,0,0])
>>> torch.nonzero(array)
tensor([[1],
        [2]])
>>> torch.where(array)
(tensor([1, 2]),)
>>> 

PS:注意nonzero和where返回tensor的shape区别。