Pytorch Tensor Object Is Not Callable