Struct PatchEmbedding 

pub struct PatchEmbedding { /* private fields */ }

Patch embedding layer for Vision Transformers

Implementations

impl PatchEmbedding

pub fn new(config: PatchEmbeddingConfig) -> Result<Self>

Create new patch embedding layer

pub fn build_patch_embed_graph(&self, graph: &mut EinsumGraph) -> Result<usize>

Build patch embedding einsum graph

Converts image patches to token embeddings

Graph Inputs

• Tensor 0: Input image tensor [batch, in_channels, height, width] or patches [batch, num_patches, patch_dim]
• Tensor 1: Patch embedding weights [patch_dim, d_model]

Graph Output

• Embedded patches [batch, num_patches, d_model]

pub fn config(&self) -> &PatchEmbeddingConfig

Get configuration