Param

Param#

class seli.net.Param(*, init: Init, rngs: Key[Array, ''] | UInt32[Array, '2'] | None = None, value: V | None = None, collection: str | None = 'param')[source]#

Bases: Module, Generic[V]

Organizes a parameter

Attributes Summary

initialized

Methods Summary

__call__(shape[, dtype])

Call self as a function.

from_value(value, *[, collection])

Attributes Documentation

initialized#

Methods Documentation

__call__(shape: tuple[int, ...], dtype: str | type[~typing.Any] | ~numpy.dtype | ~jax._src.typing.SupportsDType = <class 'jax.numpy.float32'>) → Array[source]#

Call self as a function.

classmethod from_value(value: Array, *, collection: str | None = 'param') → Param[Array][source]#
Contents