VitEntryLayers

Entry block of ViT (Patch Embed).

Abstract Signature:

VitEntryLayers(patch_size: int, input_dims: int, output_dims: int)

PyTorchPaxML / Praxis

PyTorch

API: —

Strategy: Custom / Partial

PaxML / Praxis

API: praxis.layers.VitEntryLayers

Strategy: Direct Mapping

Official Docs ↗