Struct hotg_runecoral::TensorMut

pub struct TensorMut<'a> {
    pub element_type: ElementType,
    pub buffer: &'a mut [u8],
    pub shape: Cow<'a, [c_int]>,
}

A mutable reference to a tensor’s backing buffer.

Fields

element_type: ElementType

buffer: &'a mut [u8]

shape: Cow<'a, [c_int]>

Implementations

impl<'a> TensorMut<'a>

pub fn from_slice<E: TensorElement>(
    slice: &'a mut [E],
    dimensions: &[usize]
) -> Self

Create a new TensorMut backed by a slice.

pub fn descriptor(&self) -> TensorDescriptor<'_>

Get a TensorDescriptor that describes this tensor.

Trait Implementations

impl<'a> Debug for TensorMut<'a>

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

Formats the value using the given formatter.

impl<'a> PartialEq<TensorMut<'a>> for TensorMut<'a>

fn eq(&self, other: &TensorMut<'a>) -> bool

This method tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &TensorMut<'a>) -> bool

This method tests for !=.

impl<'a> StructuralPartialEq for TensorMut<'a>