diffusion_rs_common::nn::embedding

Function embedding

source
pub fn embedding(
    in_size: usize,
    out_size: usize,
    vb: VarBuilder<'_>,
) -> Result<Embedding>