tch_plus::nn

Function linear

Source 
pub fn linear<'a, T: Borrow<Path<'a>>>(
    vs: T,
    in_dim: i64,
    out_dim: i64,
    c: LinearConfig,
) -> Linear
Expand description

Creates a new linear layer.