Momentum dependence of spin-orbit interaction effects in single-layer and multi-layer transition metal dichalcogenides