Function embedding 

pub fn embedding(
    in_size: usize,
    out_size: usize,
    vb: VarBuilderArgs<'static, ShardedSafeTensors>,
    config: &Option<QuantizedConfig>,
) -> Result<Embedding, Error>