Struct DenseLinear 

pub struct DenseLinear<B: Backend> { /* private fields */ }

Dense linear projection.

Holds a single weight matrix laid out row-major as [out_features, in_features]. forward delegates to B::gemm plus (optional) B::add_bias.

Implementations

impl<B: Backend> DenseLinear<B>

pub fn from_rows( weight_row_major: &[f32], out_features: usize, in_features: usize, ) -> Self

Build a weight-only dense projection (no bias).

pub fn from_rows_with_bias( weight_row_major: &[f32], bias: &[f32], out_features: usize, in_features: usize, ) -> Self

Build a dense projection with a bias vector of length out_features.

pub fn from_buffer( weight: B::Buffer, out_features: usize, in_features: usize, ) -> Self

Construct by moving already-allocated Backend buffers.

pub fn with_bias(self, bias: B::Buffer) -> Self

pub fn weight(&self) -> &B::Buffer

pub fn bias(&self) -> Option<&B::Buffer>

Trait Implementations

impl<B: Backend> Linear<B> for DenseLinear<B>

fn in_features(&self) -> usize

fn out_features(&self) -> usize

fn forward( &self, ctx: &mut B::Context, input: &B::Buffer, out: &mut B::Buffer, m: usize, )

Append GEMM work onto ctx. Caller flushes the context when results must be materialised.