Pytorch Create Tensor