Tensor Product In Python