今天遇到一個torch版本問題
import torch
print(torch.__version__)
a=torch.randn(10,2)
torch.min(a,axis=1)
結果如下

可以看到這個torch.min的參數axis是和torch的版本有關的
今天遇到一個torch版本問題
import torch
print(torch.__version__)
a=torch.randn(10,2)
torch.min(a,axis=1)
結果如下
可以看到這個torch.min的參數axis是和torch的版本有關的