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 0x7824acff2d90>,
- name: Optional[str] = None,
- *,
- _kd_state: 'Optional[_ModuleState]' = None,
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