AffineMappedParameter#

class hypercoil.init.mapparam.AffineMappedParameter(model: PyTree, *, loc: Tensor = 0.0, scale: Tensor = 1.0, where: Callable = <function where_weight>, image_bound: Any = None, preimage_bound: Any = None, handler: Callable = None)[source]#

Affine parameter mapper.

Methods

image_map_impl

preimage_map_impl