Enum TensorData 

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

Borrowed, contiguous tensor elements.

Variants

F32(&'a [f32])

Borrowed f32 elements.

F64(&'a [f64])

Borrowed f64 elements.

F16(&'a [f16])

Borrowed f16 elements.

BF16(&'a [bf16])

Borrowed bf16 elements.

I64(&'a [i64])

Borrowed i64 elements.

I32(&'a [i32])

Borrowed i32 elements.

I16(&'a [i16])

Borrowed i16 elements.

I8(&'a [i8])

Borrowed i8 elements.

U64(&'a [u64])

Borrowed u64 elements.

U32(&'a [u32])

Borrowed u32 elements.

U16(&'a [u16])

Borrowed u16 elements.

U8(&'a [u8])

Borrowed u8 elements.

Bool(&'a [bool])

Borrowed bool elements.

Implementations

impl TensorData<'_>

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 to_owned(self) -> TensorBuffer

Copy the elements into an owned CPU buffer.

Trait Implementations

impl<'a> Clone for TensorData<'a>

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<'a> Copy for TensorData<'a>

impl<'a> Debug for TensorData<'a>

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

Formats the value using the given formatter.