Tensor Ones Like Function In Python