Tensorflow Create Tensor Pytorch