Struct ActivationCache 

pub struct ActivationCache {
    pub tensors: Vec<Option<TensorRef>>,
}

Activation buffers retained from a forward_train call.

Implements Drop so device buffers are freed automatically.

Fields

tensors: Vec<Option<TensorRef>>

Per-node activation tensor, indexed by DAG node index. None for nodes with no cached activation (e.g. Input placeholders).

Trait Implementations

impl Drop for ActivationCache

Available on crate feature training only.

fn drop(&mut self)

Executes the destructor for this type.

fn pin_drop(self: Pin<&mut Self>)

🔬This is a nightly-only experimental API. (pin_ergonomics)

Execute the destructor for this type, but different to Drop::drop, it requires self to be pinned.