kd.nn.ZeroEmbedding

class kauldron.modules.ZeroEmbedding(

dtype: str | type[typing.Any] | numpy.dtype | jax._src.typing.SupportsDType = <class 'jax.numpy.float32'>,

parent: flax.linen.module.Module | flax.core.scope.Scope | flax.linen.module._Sentinel | None = <flax.linen.module._Sentinel object>,

name: str | None = None,

)

Bases: flax.linen.module.Module

Embedding that returns zero (for deactivating position embeddings).

Implements the knn_types.PositionEmbedding protocol.

dtype

alias of jax.numpy.float32

name: str | None = None

parent: flax.linen.module.Module | flax.core.scope.Scope | flax.linen.module._Sentinel | None = None

scope: Scope | None = None