Struct CustomOpContext 

pub struct CustomOpContext {
    pub intermediates: HashMap<String, Scirs2Tensor>,
    pub metadata: HashMap<String, String>,
    pub requires_grad: bool,
}

Context for custom operation execution.

Provides storage for intermediate values needed during backward pass and metadata about the execution environment.

Fields

intermediates: HashMap<String, Scirs2Tensor>

Storage for intermediate values (forward pass -> backward pass)

metadata: HashMap<String, String>

Custom metadata

requires_grad: bool

Whether gradient computation is enabled

Implementations

impl CustomOpContext

pub fn new() -> Self

Create a new context.

pub fn with_grad() -> Self

Create a context with gradient computation enabled.

pub fn save_for_backward( &mut self, name: impl Into<String>, tensor: Scirs2Tensor, )

Store an intermediate tensor for backward pass.

pub fn get_saved(&self, name: &str) -> Option<&Scirs2Tensor>

Retrieve a saved intermediate tensor.

pub fn set_metadata(&mut self, key: impl Into<String>, value: impl Into<String>)

Set metadata.

pub fn get_metadata(&self, key: &str) -> Option<&String>

Get metadata.

Trait Implementations

impl Clone for CustomOpContext

fn clone(&self) -> CustomOpContext

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for CustomOpContext

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

Formats the value using the given formatter.

impl Default for CustomOpContext

fn default() -> CustomOpContext

Returns the “default value” for a type.