Struct MultiOutputHandle 

pub struct MultiOutputHandle {
    pub source: NodeId,
    pub sub_shapes: Vec<Shape>,
    pub offsets: Vec<usize>,
}

Handle to a multi-output Op::CustomFn built via Graph::custom_fn_multi. Internally the op produces a flat 1-D concatenated output; this handle remembers each sub-output’s offset + original shape so Self::output can materialize component i lazily via Op::Narrow + Op::Reshape.

Fields

source: NodeId

NodeId of the wrapped Op::CustomFn (1-D F32, length = Σ sub_shapes[i].num_elements).

sub_shapes: Vec<Shape>

Original per-sub-output shapes (in declaration order).

offsets: Vec<usize>

Per-sub-output start offsets into source (element-counted, not byte-counted).

Implementations

impl MultiOutputHandle

pub fn n_outputs(&self) -> usize

Number of sub-outputs.

pub fn output(&self, g: &mut Graph, idx: usize) -> NodeId

Materialize sub-output idx as an outer-graph NodeId. Internally: Op::Narrow(source, axis=0, start=offsets[idx], len=numel(sub_shapes[idx])) → Op::Reshape back to the declared shape.

Trait Implementations

impl Clone for MultiOutputHandle

fn clone(&self) -> MultiOutputHandle

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MultiOutputHandle

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

Formats the value using the given formatter.