Struct burn_tensor::container::TensorContainer

pub struct TensorContainer<ID> { /* private fields */ }

Contains tensor of arbitrary dimension.

Implementations

impl<ID> TensorContainer<ID>where ID: Hash + PartialEq + Eq + Debug,

pub fn new() -> Self

Create an empty container.

pub fn get<B, const D: usize>(&self, id: &ID) -> Option<Tensor<B, D>>where B: Backend,

Get a tensor with the given ID.

pub fn register<B, const D: usize>(&mut self, id: ID, value: Tensor<B, D>)where B: Backend,

Register a new tensor for the given ID.

Notes

If a tensor is already registered for the given ID, it will be replaced.

pub fn remove<B, const D: usize>(&mut self, id: &ID) -> Option<Tensor<B, D>>where B: Backend,

Remove a tensor for the given ID and returns it.

pub fn len(&self) -> usize

The number of tensors registered.

pub fn is_empty(&self) -> bool

If any tensor is contained.

Trait Implementations

impl<ID: Debug> Debug for TensorContainer<ID>

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

Formats the value using the given formatter.

impl<ID> Default for TensorContainer<ID>where ID: Hash + PartialEq + Eq + Debug,

fn default() -> Self

Returns the “default value” for a type.