added batched_tensordot function