Struct GraphNode 

pub struct GraphNode {
    pub id: String,
    pub op: TensorOp,
    pub inputs: Vec<String>,
    pub output_shape: Option<Vec<usize>>,
    pub metadata: HashMap<String, String>,
}

Node in the computation graph

Fields

id: String

Unique node ID

op: TensorOp

Operation type

inputs: Vec<String>

Input node IDs

output_shape: Option<Vec<usize>>

Output shape (if known)

metadata: HashMap<String, String>

Metadata

Implementations

impl GraphNode

pub fn new(id: String, op: TensorOp) -> Self

Create a new graph node

pub fn add_input(self, input_id: String) -> Self

Add an input node

pub fn with_output_shape(self, shape: Vec<usize>) -> Self

Set output shape

pub fn add_metadata(self, key: String, value: String) -> Self

Add metadata

Trait Implementations

impl Clone for GraphNode

fn clone(&self) -> GraphNode

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GraphNode

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for GraphNode

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for GraphNode

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

where __S: Serializer,

Serialize this value into the given Serde serializer.