Get Scalar Value From Tensor Pytorch