Turn Numpy Array Into Torch Tensor Reshape