Enum ParamKind 

pub enum ParamKind {
    Weight,
    Bias,
    RunningMean,
    RunningVar,
    Raw,
}

What a leaf tensor is for. Kept coarse on purpose: it drives display grouping and the “is this trainable” question, not arithmetic.

Variants

Weight

Bias

RunningMean

RunningVar

Raw

Registered through a raw vb.get(..) / vb.get_with_hints(..) call in user code.

Trait Implementations

impl Clone for ParamKind

fn clone(&self) -> ParamKind

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for ParamKind

impl Debug for ParamKind

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

Formats the value using the given formatter.

impl Eq for ParamKind

impl PartialEq for ParamKind

fn eq(&self, other: &ParamKind) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl Serialize for ParamKind

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for ParamKind