*Easier to understand interface, less code* - if model assumes 4d input, put into model ensure_4d or something like that - remove code everywhere posible that adds empty dimensions etc.
Easier to understand interface, less code