[Kwant] Transmission matrix and momenta