If `h` should be optimised, one should add a function in `trainable` and set `h` as a vector type field.