Hi, I am trying to register a possibly None parameter into a HybridBlock model, so that whenever it is passed to hybrid_forward() automatically through model._reg_params, the function can do some stuff based on whether the parameter is None. I tried to use model.params.get() and model.params.get_constant(), but it seems none of them can have the parameter taking value None when initializing the parameter. Is there any way to do this smartly? Like in PyTorch, this can be done by model.register_parameter(‘empty_param’, None), after which the ‘empty_param’ will appear in model.parameters() iterator list.
I am not sure I fully understand your question. Can you give a pseudo-code example of what you are trying to implement? Why would an initialization to zero values instead of None not work?