InitTruncatedNormal#

class seli.net.InitTruncatedNormal(minv: float = -1.0, maxv: float = 1.0, shift: float = 0.0, scale: float = 1.0)[source]#

Bases: Init

Initializes values from a truncated normal distribution.

Methods Summary

__call__(key, shape, dtype)

Call self as a function.

Methods Documentation

__call__(key: Key[Array, ''] | UInt32[Array, '2'], shape: tuple[int, ...], dtype: str | type[Any] | dtype | SupportsDType) Array[source]#

Call self as a function.