今天遇到一個torch版本問題
import torch
print(torch.__version__)
a=torch.randn(10,2)
torch.min(a,axis=1)
結果如下
![](https://img.laitimes.com/img/9ZDMuAjOiMmIsIjOiQnIsIyZuBnLzEWZiZTM5QjNlVWYxEDMhNjNwQDM1M2Y3kjNlVDOwEzLc52YucWbp5GZzNmLn9Gbi1yZtl2Lc9CX6MHc0RHaiojIsJye.png)
可以看到這個torch.min的參數axis是和torch的版本有關的
今天遇到一個torch版本問題
import torch
print(torch.__version__)
a=torch.randn(10,2)
torch.min(a,axis=1)
結果如下
可以看到這個torch.min的參數axis是和torch的版本有關的