* Implement `L_Op` for `join_dims` and `split_dims`

Improve type hints for `join_dims` and `split_dims`

* Feedback