max関数(torch)