Enum Tensor 

pub enum Tensor {
    F32(HostTensor<f32>),
    F64(HostTensor<f64>),
    I32(HostTensor<i32>),
    I64(HostTensor<i64>),
    Bool(HostTensor<bool>),
    C32(HostTensor<Complex32>),
    C64(HostTensor<Complex64>),
}

Dynamic owned host tensor over the supported dtype set.

Examples

use tenferro_tensor_core::{DType, Tensor};

let tensor = Tensor::from_vec_col_major(vec![2], vec![1.0_f64, 2.0])?;
assert_eq!(tensor.dtype(), DType::F64);

Variants

F32(HostTensor<f32>)

F64(HostTensor<f64>)

I32(HostTensor<i32>)

I64(HostTensor<i64>)

Bool(HostTensor<bool>)

C32(HostTensor<Complex32>)

C64(HostTensor<Complex64>)

Implementations

impl Tensor

pub fn from_vec_col_major<T: TensorScalar>( shape: impl Into<ShapeVec>, data: Vec<T>, ) -> Result<Self>

Create a dynamic tensor from a column-major host buffer.

Examples

use tenferro_tensor_core::{DType, Tensor};

let tensor = Tensor::from_vec_col_major(vec![1], vec![2.0_f32])?;
assert_eq!(tensor.dtype(), DType::F32);

pub fn dtype(&self) -> DType

Return the tensor dtype tag.

Examples

use tenferro_tensor_core::{DType, Tensor};

let tensor = Tensor::from_vec_col_major(vec![1], vec![false])?;
assert_eq!(tensor.dtype(), DType::Bool);

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

Borrow the tensor shape.

Examples

use tenferro_tensor_core::Tensor;

let tensor = Tensor::from_vec_col_major(vec![2], vec![1_i32, 2])?;
assert_eq!(tensor.shape(), &[2]);

pub fn rank(&self) -> usize

Return the tensor rank.

Examples

use tenferro_tensor_core::Tensor;

let tensor = Tensor::from_vec_col_major(vec![1, 1], vec![1_i64])?;
assert_eq!(tensor.rank(), 2);

pub fn is_empty(&self) -> bool

Return whether the tensor has zero elements.

Examples

use tenferro_tensor_core::Tensor;

let tensor = Tensor::from_vec_col_major(vec![0], Vec::<f64>::new())?;
assert!(tensor.is_empty());

pub fn as_slice<T: TensorScalar>(&self) -> Result<&[T]>

Borrow the typed host slice when the dtype matches.

Examples

use tenferro_tensor_core::Tensor;

let tensor = Tensor::from_vec_col_major(vec![1], vec![3.0_f64])?;
assert_eq!(tensor.as_slice::<f64>()?, &[3.0]);
assert!(tensor.as_slice::<f32>().is_err());

pub fn as_mut_slice<T: TensorScalar>(&mut self) -> Result<&mut [T]>

Mutably borrow the typed host slice when the dtype matches.

Examples

use tenferro_tensor_core::Tensor;

let mut tensor = Tensor::from_vec_col_major(vec![1], vec![3.0_f64])?;
tensor.as_mut_slice::<f64>()?[0] = 4.0;
assert_eq!(tensor.as_slice::<f64>()?, &[4.0]);

pub fn as_view(&self) -> TensorView<'_>

Borrow this tensor as a dynamic zero-offset view.

Examples

use tenferro_tensor_core::{DType, Tensor};

let tensor = Tensor::from_vec_col_major(vec![1], vec![1_i64])?;
assert_eq!(tensor.as_view().dtype(), DType::I64);

pub fn into_vec_col_major<T: TensorScalar>(self) -> Result<(ShapeVec, Vec<T>)>

Consume this tensor and return typed column-major data when the dtype matches.

Examples

use tenferro_tensor_core::Tensor;

let tensor = Tensor::from_vec_col_major(vec![1], vec![2.0_f32])?;
assert_eq!(tensor.into_vec_col_major::<f32>()?.1, vec![2.0]);

Trait Implementations

impl Clone for Tensor

fn clone(&self) -> Tensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Tensor

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

Formats the value using the given formatter.

impl PartialEq for Tensor

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