code: def reg_to_mx(layer_type) -> FunctionType:
      """function decorator that registers layer implementation
      :return: decorated function
      """
  
      def reg_impl_decorator(func):
          assert layer_type not in mx_layer_dict
          mx_layer_dict[layer_type] = func
          return func
  
      return reg_impl_decorator
  
imports: from types import FunctionType
  from mxnet.gluon import nn
  from rl_coach.architectures import layers
  from rl_coach.architectures.mxnet_components import utils
label_desc: Use image node_id 1 for calling a global function with example usage: reg_to_mx(layer_type) and returns: reg_impl_decorator, func