Struct GraphScope 

pub struct GraphScope { /* private fields */ }

Owns a traced graph while you build symbolic tensors.

Implementations

impl GraphScope

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

pub fn graph(&self) -> Graph

pub fn fresh_symbol(&mut self) -> u32

Fresh dynamic-dimension symbol (for variable batch / sequence length).

pub fn input( &mut self, name: impl Into<String>, shape: impl Into<Shape>, ) -> Tensor

pub fn input_dyn( &mut self, name: impl Into<String>, dims: &[Dim], dtype: DType, ) -> Tensor

Input with a caller-chosen dynamic symbol on one axis.

pub fn param( &mut self, name: impl Into<String>, shape: impl Into<Shape>, ) -> Tensor

pub fn scalar(&mut self, value: f64) -> Tensor

Rank-0 literal broadcastable in binary ops.

pub fn constant(&mut self, value: f64, dtype: DType) -> Tensor

pub fn cat(&mut self, tensors: &[&Tensor], axis: usize) -> Tensor

Stack existing tensors along axis (zero-copy at IR level).

pub fn set_outputs( &mut self, outputs: impl IntoIterator<Item = impl Into<NodeId>>, )

pub fn finish(self) -> Graph

Trait Implementations

impl Debug for GraphScope

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

Formats the value using the given formatter.