Struct Tensor 

pub struct Tensor { /* private fields */ }

Implementations

impl Tensor

pub fn get_type(&self) -> DataType

Creates a new tensor with the specified data type.

pub unsafe fn set_type(&mut self, kind: DataType)

Sets the data type of the tensor.

Arguments

• kind: The data type of the tensor.

Examples

use airs_types::Tensor;
let mut tensor = Tensor::default();
unsafe { tensor.set_type(airs_types::DataType::Float) };

pub fn cast_type(&mut self, kind: DataType)

pub fn with_type(self, kind: DataType) -> Self

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

pub unsafe fn set_bytes(&mut self, bytes: &[u8])

pub fn with_bytes(self, bytes: &[u8]) -> Self

Trait Implementations

impl Debug for Tensor

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

Formats the value using the given formatter.

impl Default for Tensor

fn default() -> Tensor

Returns the “default value” for a type.

impl From<&[ImageBuffer<Rgb<u8>, Vec<u8>>]> for Tensor

fn from(value: &[RgbImage]) -> Self

Converts to this type from the input type.