问题:
../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区别。