Set number of hidden layers separately for each element

[hghcomphys]

This can be done via element_nodes_short keyword.
This will reduce number of trainable parameters and therefore improving the performance.