kd.knn.Dropout

kd.knn.Dropout#

class kauldron.klinen.Dropout(
rate: float,
broadcast_dims: collections.abc.Sequence[int] = (),
deterministic: bool | None = None,
rng_collection: str = 'dropout',
parent: Union[flax.linen.module.Module,
flax.core.scope.Scope,
flax.linen.module._Sentinel,
NoneType] = <flax.linen.module._Sentinel object at 0x76410688a690>,
name: Optional[str] = None,
*,
_kd_state: 'Optional[_ModuleState]' = None,
)[source]

Bases: flax.linen.stochastic.Dropout, kauldron.klinen.module.Module

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