首页 > itarticle > torch.max()用法

torch.max()用法

admin 11月 12, 2020 0

1
2
3

tensor = 样本数*属性数
torch.max(tensor,1) 
torch.max(tensor,0) #在每个属性值中最大值的属性值及其索引

近期文章