pytorch 常用函数 max ,eq