kd.nn.Patchify#
- class kauldron.modules.Patchify(patch_size: tuple[int, int], parent: flax.linen.module.Module | flax.core.scope.Scope | flax.linen.module._Sentinel | None = <flax.linen.module._Sentinel object>, name: str | None = None)[source]
Bases:
flax.linen.module.ModulePatchify an image, as in ViT (without linear embedding).
- patch_size: tuple[int, int]
- name: str | None = None
- parent: flax.linen.module.Module | flax.core.scope.Scope | flax.linen.module._Sentinel | None = None
- scope: Scope | None = None