Struct SparseSgdTable 

pub struct SparseSgdTable { /* private fields */ }

Sparse SGD optimizer over a 2-D embedding table (CSR gradients).

Implementations

impl SparseSgdTable

pub fn new(config: SparseSgdConfig) -> Self

Create a new table optimizer.

pub fn step( &mut self, params: &mut Array2<f64>, grad: &CsrGradient, ) -> Result<(), GpuOptimError>

Perform one sparse SGD update on the embedding table in place.

Errors

Returns GpuOptimError::DimensionMismatch on shape disagreement and propagates gradient validation errors.

Trait Implementations

impl Clone for SparseSgdTable

fn clone(&self) -> SparseSgdTable

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for SparseSgdTable

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.