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 速查表

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章