Struct Tensor 

pub struct Tensor { /* private fields */ }

A tensor holding raw bytes with data type and shape metadata.

This represents a multidimensional array of fixed-size elements in C-contiguous (row-major) order.

Implementations

impl Tensor

pub fn new( bytes: impl Into<ArrayBytesRaw<'static>>, data_type: DataType, shape: Vec<u64>, ) -> Self

Create a new Tensor.

pub fn bytes(&self) -> &[u8]

Get the raw bytes.

pub fn data_type(&self) -> &DataType

Get the data type.

pub fn shape(&self) -> &[u64]

Get the shape.

pub fn into_parts(self) -> (Cow<'static, [u8]>, DataType, Vec<u64>)

Consume self and return the parts.

pub fn as_parts(&self) -> (&[u8], &DataType, &[u64])

Get references to all parts of the tensor.

Returns (bytes, data_type, shape) as references.

Trait Implementations

impl FromArrayBytes for Tensor

fn from_array_bytes( bytes: ArrayBytes<'static>, shape: &[u64], data_type: &DataType, ) -> Result<Self, ArrayError>

Convert ArrayBytes into Self.

fn from_array_bytes_arc( bytes: Arc<ArrayBytes<'static>>, shape: &[u64], data_type: &DataType, ) -> Result<Self, ArrayError>

Convert an Arc<ArrayBytes> into Self.

impl<'a> IntoArrayBytes<'a> for &'a Tensor

fn into_array_bytes( self, data_type: &DataType, ) -> Result<ArrayBytes<'a>, ElementError>

Convert self into ArrayBytes.

impl IntoArrayBytes<'static> for Tensor

fn into_array_bytes( self, data_type: &DataType, ) -> Result<ArrayBytes<'static>, ElementError>

Convert self into ArrayBytes.

impl TensorLike<RowMajorCompactLayout> for Tensor

Available on crate feature dlpack only.

type Error = TensorError

fn data_ptr(&self) -> *mut c_void

fn memory_layout(&self) -> RowMajorCompactLayout

fn byte_offset(&self) -> u64

fn device(&self) -> Result<Device, Self::Error>

fn data_type(&self) -> Result<DataType, Self::Error>