VitEntryLayers ============== Entry block of ViT (Patch Embed). **Abstract Signature:** ``VitEntryLayers(patch_size: int, input_dims: int, output_dims: int)`` .. raw:: html

PyTorch

API:
Strategy: Custom / Partial

PaxML / Praxis

API: praxis.layers.VitEntryLayers
Strategy: Direct Mapping