pytorch常用函数总结(持续更新) torch.max(input,dim) 求取指定维度上的最大值,,返回输入张量给定维度上每行的最大值,并同时返回每个最大值的位置索引.比如: demo.shape Out[7]: torch.Size([10, 3, 10, 10]) torch.max(demo,1)[0].shape Out[8]: torch.Size([10, 10, 10]) torch.max(demo,1)[0]这其中的[0]取得就是返回的最大值,torch.max(dem…