Enum TensorBuffer 

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

Owned, contiguous tensor elements stored on the CPU.

Variants

F32(Vec<f32>)

Owned f32 elements.

F64(Vec<f64>)

Owned f64 elements.

F16(Vec<f16>)

Owned f16 elements.

BF16(Vec<bf16>)

Owned bf16 elements.

I64(Vec<i64>)

Owned i64 elements.

I32(Vec<i32>)

Owned i32 elements.

I16(Vec<i16>)

Owned i16 elements.

I8(Vec<i8>)

Owned i8 elements.

U64(Vec<u64>)

Owned u64 elements.

U32(Vec<u32>)

Owned u32 elements.

U16(Vec<u16>)

Owned u16 elements.

U8(Vec<u8>)

Owned u8 elements.

Bool(Vec<bool>)

Owned bool elements.

Implementations

impl TensorBuffer

pub fn dtype(&self) -> DType

Element type of this buffer.

pub fn len(&self) -> usize

Number of elements, not bytes.

pub fn is_empty(&self) -> bool

Whether the buffer contains no elements.

pub fn view(&self) -> TensorData<'_>

Borrow the elements without copying.

pub fn view_mut(&mut self) -> TensorDataMut<'_>

Borrow the elements for mutation without copying.

Trait Implementations

impl Clone for TensorBuffer

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorBuffer

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

Formats the value using the given formatter.

impl PartialEq for TensorBuffer

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for TensorBuffer