Get Max Value From Tensor Pytorch Github