Function tch::nn::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.