Struct MixedOperation 

pub struct MixedOperation {
    pub arch_params: Vec<f64>,
    pub operation_outputs: Option<Vec<Vec<f64>>>,
}

One mixed operation on a directed edge in the DARTS cell DAG.

Maintains per-operation un-normalised log-weights (architecture parameters).

Fields

arch_params: Vec<f64>

Un-normalised architecture parameters α_k, one per operation.

operation_outputs: Option<Vec<Vec<f64>>>

Cached per-operation outputs from the last forward call.

Implementations

impl MixedOperation

pub fn new(n_ops: usize) -> Self

Create a new MixedOperation with n_ops operations, initialised to uniform architecture weights (all log-weights = 0).

pub fn weights(&self, temperature: f64) -> Vec<f64>

Compute softmax-normalised operation weights at the given temperature.

weights[k] = exp(α_k / T) / Σ_j exp(α_j / T)

pub fn forward( &mut self, x: &[f64], op_fn: impl Fn(usize, &[f64]) -> Vec<f64>, temperature: f64, ) -> Vec<f64>

Forward pass: weighted sum Σ_k w_k · op_k(x).

op_fn(k, x) returns the output of operation k applied to input x.

pub fn argmax_op(&self) -> usize

Index of the operation with the highest architecture weight.

Trait Implementations

impl Clone for MixedOperation

fn clone(&self) -> MixedOperation

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MixedOperation

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

Formats the value using the given formatter.