Struct Context 

pub struct Context {
    pub saved_tensors: Vec<Box<Tensor>>,
}

Represents the computation context, storing tensors for backward pass computations.

Fields

saved_tensors: Vec<Box<Tensor>>

Implementations

impl Context

pub fn new() -> Context

Creates a new Context instance.

Returns

(Context): A new Context instance.

Examples

use tinygrad::Context;

let context = Context::new();

pub fn save_for_backward(&mut self, tensors: Vec<Box<Tensor>>)

Saves tensors for backward pass.

Arguments

• tensors - A vector of Boxed Tensors to be saved.

Examples

use ndarray::{array, Array1};
use tinygrad::{Tensor, Context};

let mut ctx = Context::new();
let tensor = Tensor::new(array![1.0, 2.0, 3.0]);
ctx.save_for_backward(vec![Box::new(tensor)]);

Trait Implementations

impl Default for Context

fn default() -> Self

Returns the “default value” for a type.