Struct TreeNode 

pub struct TreeNode {
    pub name: String,
    pub node_type: String,
    pub children: Vec<TreeNode>,
    pub shape: Option<Vec<usize>>,
    pub dtype: Option<String>,
}

Node in model hierarchy tree

Fields

name: String

Node name

node_type: String

Node type

children: Vec<TreeNode>

Child nodes

shape: Option<Vec<usize>>

Tensor shape (if leaf node)

dtype: Option<String>

Data type (if leaf node)

Implementations

impl TreeNode

pub fn new(name: impl Into<String>, node_type: impl Into<String>) -> Self

Create new tree node

pub fn tensor( name: impl Into<String>, shape: Vec<usize>, dtype: impl Into<String>, ) -> Self

Create leaf node (tensor)

pub fn add_child(&mut self, child: TreeNode)

Add child node

pub fn count_nodes(&self) -> usize

Count total nodes

Trait Implementations

impl Clone for TreeNode

fn clone(&self) -> TreeNode

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TreeNode

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

Formats the value using the given formatter.