Turn Numpy Array Into Torch Tensor Gradient