Struct ViTBlockConfig 

pub struct ViTBlockConfig {
    pub embed_dim: usize,
    pub n_heads: usize,
    pub mlp_ratio: usize,
}

Configuration for a single ViT transformer block.

Fields

embed_dim: usize

Token embedding dimension.

n_heads: usize

Number of attention heads. Must divide embed_dim.

mlp_ratio: usize

MLP hidden-dim multiplier: mlp_dim = mlp_ratio * embed_dim.

Implementations

impl ViTBlockConfig

pub fn new( embed_dim: usize, n_heads: usize, mlp_ratio: usize, ) -> VisionResult<Self>

Create and validate a ViTBlockConfig.

Errors

• embed_dim == 0 → InvalidEmbedDim
• n_heads == 0 → InvalidNumHeads
• embed_dim % n_heads != 0 → HeadDimMismatch

pub fn head_dim(&self) -> usize

Dimension per attention head.

pub fn mlp_dim(&self) -> usize

MLP hidden dimension.

Trait Implementations

impl Clone for ViTBlockConfig

fn clone(&self) -> ViTBlockConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ViTBlockConfig

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

Formats the value using the given formatter.

impl PartialEq for ViTBlockConfig

fn eq(&self, other: &ViTBlockConfig) -> 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 ViTBlockConfig