register

register(klass)

Registers a custom initializer.

Custom initializers can be created by extending mx.init.Initializer and implementing the required functions like _init_weight and _init_bias. The created initializer must be registered using mx.init.register before it can be called by name.

Parameters:

klass : class

A subclass of mx.init.Initializer that needs to be registered as a custom initializer.