Enum TensorData

pub enum TensorData {
    U8(Buffer<u8>),
    U16(Buffer<u16>),
    U32(Buffer<u32>),
    U64(Buffer<u64>),
    I8(Buffer<i8>),
    I16(Buffer<i16>),
    I32(Buffer<i32>),
    I64(Buffer<i64>),
    F16(Buffer<f16>),
    F32(Buffer<f32>),
    F64(Buffer<f64>),
    JPEG(Buffer<u8>),
}

Flattened Tensor data payload

Examples

assert_eq!(
    TensorData::data_type(),
    DataType::Union(
        vec![
            Field::new("U8", DataType::Binary, false),
            Field::new(
                "U16",
                DataType::List(Box::new(Field::new("item", DataType::UInt16, false))),
                false
            ),
            Field::new(
                "U32",
                DataType::List(Box::new(Field::new("item", DataType::UInt32, false))),
                false
            ),
            Field::new(
                "U64",
                DataType::List(Box::new(Field::new("item", DataType::UInt64, false))),
                false
            ),
            Field::new(
                "I8",
                DataType::List(Box::new(Field::new("item", DataType::Int8, false))),
                false
            ),
            Field::new(
                "I16",
                DataType::List(Box::new(Field::new("item", DataType::Int16, false))),
                false
            ),
            Field::new(
                "I32",
                DataType::List(Box::new(Field::new("item", DataType::Int32, false))),
                false
            ),
            Field::new(
                "I64",
                DataType::List(Box::new(Field::new("item", DataType::Int64, false))),
                false
            ),
            Field::new(
                "F16",
                DataType::List(Box::new(Field::new("item", DataType::Float16, false))),
                false
            ),
            Field::new(
                "F32",
                DataType::List(Box::new(Field::new("item", DataType::Float32, false))),
                false
            ),
            Field::new(
                "F64",
                DataType::List(Box::new(Field::new("item", DataType::Float64, false))),
                false
            ),
            Field::new("JPEG", DataType::Binary, false),
        ],
        None,
        UnionMode::Dense
    ),
);

Variants

U8(Buffer<u8>)

U16(Buffer<u16>)

U32(Buffer<u32>)

U64(Buffer<u64>)

I8(Buffer<i8>)

I16(Buffer<i16>)

I32(Buffer<i32>)

I64(Buffer<i64>)

F16(Buffer<f16>)

F32(Buffer<f32>)

F64(Buffer<f64>)

JPEG(Buffer<u8>)

Implementations

impl TensorData

pub fn dtype(&self) -> TensorDataType

pub fn size_in_bytes(&self) -> usize

pub fn is_empty(&self) -> bool

pub fn is_compressed_image(&self) -> bool

Trait Implementations

impl ArrowDeserialize for TensorData

type ArrayType = TensorDataArray

The arrow2::Array type corresponding to this field

fn arrow_deserialize<'a>(v: Option<Self>) -> Option<Self>

Deserialize this field from arrow

impl ArrowField for TensorData

type Type = TensorData

This should be Self except when implementing large offset and fixed placeholder types. For the later, it should refer to the actual type. For example when the placeholder type is LargeString, this should be String.

fn data_type() -> DataType

The DataType

impl ArrowSerialize for TensorData

type MutableArrayType = MutableTensorDataArray

The arrow2::array::MutableArray that holds this value

fn new_array() -> Self::MutableArrayType

Create a new mutable array

fn arrow_serialize(v: &Self, array: &mut Self::MutableArrayType) -> Result<()>

Serialize this field to arrow

impl Clone for TensorData

fn clone(&self) -> TensorData

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for TensorData

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

Formats the value using the given formatter.

impl PartialEq for TensorData

fn eq(&self, other: &TensorData) -> 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 StructuralPartialEq for TensorData