e3x.nn.initializers.FusedTensorInitializerFn

class e3x.nn.initializers.FusedTensorInitializerFn(*args, **kwargs)[source]

Bases: Protocol

Protocol for fused tensor initializer functions.

__call__(scale=1.0, mask=True, dtype=<class 'jax.numpy.float64'>)[source]
Return type:

InitializerFn