天天看點

torch.min 錯誤問題

今天遇到一個torch版本問題

import torch
print(torch.__version__)
a=torch.randn(10,2)
torch.min(a,axis=1)
           

結果如下

torch.min 錯誤問題
torch.min 錯誤問題

可以看到這個torch.min的參數axis是和torch的版本有關的

繼續閱讀