Struct PatchEmbeddingConfig 

pub struct PatchEmbeddingConfig {
    pub in_channels: usize,
    pub patch_h: usize,
    pub patch_w: usize,
    pub embed_dim: usize,
}

Configuration for patch embedding.

Example

use jepa_vision::patch::PatchEmbeddingConfig;
use burn_ndarray::NdArray;
use burn::prelude::*;

type B = NdArray<f32>;
let device = burn_ndarray::NdArrayDevice::Cpu;

let config = PatchEmbeddingConfig::new(3, 16, 16, 256);
let patch_embed = config.init::<B>(&device);
assert_eq!(patch_embed.num_patches(224, 224), 196);

Fields

in_channels: usize

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

patch_h: usize

Patch height in pixels.

patch_w: usize

Patch width in pixels.

embed_dim: usize

Output embedding dimension.

Implementations

impl PatchEmbeddingConfig

pub fn new( in_channels: usize, patch_h: usize, patch_w: usize, embed_dim: usize, ) -> Self

Create a new config with the given parameters.

pub fn init<B: Backend>(&self, device: &B::Device) -> PatchEmbedding<B>

Initialize a PatchEmbedding module.

Trait Implementations

impl Clone for PatchEmbeddingConfig

fn clone(&self) -> PatchEmbeddingConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PatchEmbeddingConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for PatchEmbeddingConfig

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for PatchEmbeddingConfig

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.