Function linear

pub fn linear(
    in_dim: usize,
    out_dim: usize,
    vb: VarBuilderArgs<'_, ShardedSafeTensors<'_>>,
) -> Result<Linear, Error>