Get Max Value From Tensor Pytorch Vs Tensorflow