diffusion_rs_common::nn::linear

Function linear

source
pub fn linear(
    in_dim: usize,
    out_dim: usize,
    vb: VarBuilder<'_>,
) -> Result<Linear>
Expand description

Create or initialize a new linear layer.

This uses some default names for weights and biases, namely "weight" and "bias".