Struct PatchEmbedConfig 

pub struct PatchEmbedConfig {
    pub img_size: usize,
    pub patch_size: usize,
    pub in_chans: usize,
    pub embed_dim: usize,
}

Configuration for the patch embedder.

The image is assumed to have shape [in_chans, img_size, img_size]. It is split into (img_size / patch_size)² non-overlapping patches, each of size in_chans × patch_size × patch_size, which are projected linearly to embed_dim.

Fields

img_size: usize

Square image spatial dimension (H = W).

patch_size: usize

Patch spatial dimension (P × P windows, stride = P).

in_chans: usize

Number of input channels (e.g., 3 for RGB).

embed_dim: usize

Output token embedding dimension.

Implementations

impl PatchEmbedConfig

pub fn new( img_size: usize, patch_size: usize, in_chans: usize, embed_dim: usize, ) -> VisionResult<Self>

Create and validate a PatchEmbedConfig.

pub fn grid_size(&self) -> usize

Number of patches along one spatial dimension.

pub fn n_patches(&self) -> usize

Total number of patches (CLS token not counted here).

pub fn kernel_vol(&self) -> usize

Kernel volume for one filter: in_chans × patch_size².

Trait Implementations

impl Clone for PatchEmbedConfig

fn clone(&self) -> PatchEmbedConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for PatchEmbedConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl PartialEq for PatchEmbedConfig

fn eq(&self, other: &PatchEmbedConfig) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for PatchEmbedConfig