Contents

struct

SushiAI::Optim::ParameterSlot

One parameter and its gradient, as the optimizer sees them.

Declared in
include/SushiAI/optim/optimizer.hpp

Tensors by value: a SushiBLAS::Tensor is a handle that shares its Storage, so a copy costs a reference-count bump and removes any dependence on where the pool keeps its map entries.

Public attributes

std::string name

The parameter's dotted name, e.g.

"0.weight".

SushiBLAS::Tensor parameter

The parameter itself; updated in place.

SushiBLAS::Tensor gradient

Its gradient, rewritten by every step.