天天看点

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的版本有关的

继续阅读