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