fix tensordot arguments