Struct burn::tensor::repr::TensorDescription

pub struct TensorDescription {
    pub id: TensorId,
    pub shape: Vec<usize>,
    pub status: TensorStatus,
    pub dtype: DType,
}

A tensor definition represents a snapshot of a tensor when it was used.

Example

A tensor that is used multiple times has its status updated for each operation.

1. Status::NotInit
2. Status::ReadOnly
3. Status::ReadOnly
4. Status::ReadWrite

Fields

id: TensorId

The tensor id.

shape: Vec<usize>

The shape of the tensor.

status: TensorStatus

The status of the tensor when it was used.

dtype: DType

The type of the tensor.

Trait Implementations

impl Clone for TensorDescription

fn clone(&self) -> TensorDescription

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for TensorDescription

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for TensorDescription

fn deserialize<__D>( __deserializer: __D, ) -> Result<TensorDescription, <__D as Deserializer<'de>>::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Hash for TensorDescription

fn hash<__H>(&self, state: &mut __H)

where __H: Hasher,

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for TensorDescription

fn eq(&self, other: &TensorDescription) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Serialize for TensorDescription

fn serialize<__S>( &self, __serializer: __S, ) -> Result<<__S as Serializer>::Ok, <__S as Serializer>::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Eq for TensorDescription

impl StructuralPartialEq for TensorDescription