Enum TensorDataMut 

pub enum TensorDataMut<'a> {
    F32(&'a mut [f32]),
    F64(&'a mut [f64]),
    F16(&'a mut [f16]),
    BF16(&'a mut [bf16]),
    I64(&'a mut [i64]),
    I32(&'a mut [i32]),
    I16(&'a mut [i16]),
    I8(&'a mut [i8]),
    U64(&'a mut [u64]),
    U32(&'a mut [u32]),
    U16(&'a mut [u16]),
    U8(&'a mut [u8]),
    Bool(&'a mut [bool]),
}

Exclusively borrowed, contiguous tensor elements.

Variants

F32(&'a mut [f32])

Mutable f32 elements.

F64(&'a mut [f64])

Mutable f64 elements.

F16(&'a mut [f16])

Mutable f16 elements.

BF16(&'a mut [bf16])

Mutable bf16 elements.

I64(&'a mut [i64])

Mutable i64 elements.

I32(&'a mut [i32])

Mutable i32 elements.

I16(&'a mut [i16])

Mutable i16 elements.

I8(&'a mut [i8])

Mutable i8 elements.

U64(&'a mut [u64])

Mutable u64 elements.

U32(&'a mut [u32])

Mutable u32 elements.

U16(&'a mut [u16])

Mutable u16 elements.

U8(&'a mut [u8])

Mutable u8 elements.

Bool(&'a mut [bool])

Mutable bool elements.

Implementations

impl TensorDataMut<'_>

pub fn dtype(&self) -> DType

Element type of this slice.

pub fn len(&self) -> usize

Number of elements, not bytes.

pub fn is_empty(&self) -> bool

Whether the slice contains no elements.

pub fn copy_from(&mut self, source: TensorData<'_>) -> Result<()>

Copy elements after checking that types and lengths match.

Trait Implementations

impl<'a> Debug for TensorDataMut<'a>

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

Formats the value using the given formatter.