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() -> TensorContainer<ID>

Create an empty container.

pub fn get<B>(&self, id: &ID) -> Option<TensorPrimitive<B>>

where B: Backend,

Get a tensor with the given ID.

pub fn register<B>(&mut self, id: ID, value: TensorPrimitive<B>)

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>(&mut self, id: &ID) -> Option<TensorPrimitive<B>>

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 for TensorContainer<ID>

where ID: Debug,

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

Formats the value using the given formatter.

impl<ID> Default for TensorContainer<ID>

where ID: Hash + PartialEq + Eq + Debug,

fn default() -> TensorContainer<ID>

Returns the “default value” for a type.