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