AttributeError: 'Tensor' object has no attribute 'argsort'

image_pred = image_pred[(-score).argsort()]

yolov3程序调试时出错:pytorch的tensor没有argsort方法

不希望转成numpy再转回tensor,查阅手册发现 pytorch的sort同时返回了sorted和indices,代码修改为:

srtd, indices = torch.sort(-score) 
image_pred = image_pred[indices]

附上numpy和pytorch对照表,来自pytorch中文网:
面向 Numpy 用户的 PyTorch 速查表

你可能感兴趣的:(pytorch)