今天遇到一个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的版本有关的