Struct DartsCell 

pub struct DartsCell {
    pub n_nodes: usize,
    pub n_input_nodes: usize,
    pub edges: Vec<Vec<MixedOperation>>,
}

A DARTS cell: a DAG with n_input_nodes fixed inputs and n_nodes intermediate nodes. Each node aggregates outputs from all prior nodes via MixedOperation edges.

Fields

n_nodes: usize

Number of intermediate (learnable) nodes.

n_input_nodes: usize

Number of fixed input nodes (typically 2 in the standard DARTS setup).

edges: Vec<Vec<MixedOperation>>

edges[i][j] is the MixedOperation from node j to intermediate node i. Node indices: 0..n_input_nodes are inputs; n_input_nodes..n_input_nodes+n_nodes are intermediate nodes.

Implementations

impl DartsCell

pub fn new( n_input_nodes: usize, n_intermediate_nodes: usize, n_ops: usize, ) -> Self

Create a new DARTS cell.

Arguments

• n_input_nodes: Number of input nodes (preceding-cell outputs).
• n_intermediate_nodes: Number of intermediate nodes to build.
• n_ops: Number of candidate operations per edge.

pub fn forward(&mut self, inputs: &[Vec<f64>], temperature: f64) -> Vec<f64>

Forward pass through the cell.

Each intermediate node output = Σ_{j < i} mixed_op_{ij}(node_j_output). Final cell output = concatenation of all intermediate node outputs.

Arguments

• inputs: Outputs of the n_input_nodes preceding this cell.
• temperature: Softmax temperature forwarded to each MixedOperation.

pub fn arch_parameters(&self) -> Vec<f64>

Collect all architecture parameters from every edge in this cell, flattened.

pub fn update_arch_params( &mut self, grads: &[f64], lr: f64, ) -> OptimizeResult<()>

Apply gradient updates to architecture parameters using a gradient slice.

grads must have the same length as arch_parameters().

pub fn derive_discrete(&self) -> Vec<Vec<usize>>

Derive the discrete architecture for this cell: argmax per edge.

Returns a Vec<Vec<usize>> with the same shape as edges, where each entry is the index of the best operation.

impl DartsCell

pub fn set_arch_params(&mut self, params: &[f64]) -> OptimizeResult<()>

Replace all architecture parameters with values from the flat slice.

Trait Implementations

impl Clone for DartsCell

fn clone(&self) -> DartsCell

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DartsCell

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

Formats the value using the given formatter.