pytorch中的nonzero()函数

报错描述

UserWarning: This overload of nonzero is deprecated:
nonzero()
Consider using one of the following signatures instead:
nonzero(*, bool as_tuple) (Triggered internally at …\torch\csrc\utils\python_arg_parser.cpp:766.)
ids = (ovr <= threshold).nonzero().squeeze()

修改

报错描述中写得比较清楚，nonzero()函数需要加入bool as_tuple 参数。

原错误语句为：ids = (ovr <= threshold).nonzero().squeeze()
改为：ids = (ovr <= threshold).nonzero(as_tuple=False).squeeze()

欢迎关注【OAOA】