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

image_pred = image_pred[(-score).argsort()] yolov3程序調試時出錯:pytorch的tensor沒有argsort方法python 不但願轉成numpy再轉回tensor,查閱手冊發現 pytorch的sort同時返回了sorted和indices,代碼修改成:web srtd, indices = torch.sort(-score) image
相關文章
相關標籤/搜索