torch.max()

https://blog.csdn.net/Z_lbj/article/details/79766690