Enum TensorRef 

pub enum TensorRef<'a> {
    Tensor(&'a Tensor),
    View(TensorView<'a>),
}

Core-neutral tensor input reference.

Examples

use tenferro_tensor_core::{Tensor, TensorRef};

let tensor = Tensor::from_vec_col_major(vec![1], vec![1.0_f32])?;
let reference = TensorRef::Tensor(&tensor);
assert_eq!(reference.shape(), &[1]);

Variants

Tensor(&'a Tensor)

View(TensorView<'a>)

Implementations

impl<'a> TensorRef<'a>

pub fn dtype(&self) -> DType

Return the referenced dtype.

Examples

use tenferro_tensor_core::{DType, Tensor, TensorRef};

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

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

Borrow the referenced shape.

Examples

use tenferro_tensor_core::{Tensor, TensorRef};

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

pub fn rank(&self) -> usize

Return the referenced rank.

Examples

use tenferro_tensor_core::{Tensor, TensorRef};

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

pub fn is_empty(&self) -> bool

Return whether the referenced tensor/view is empty.

Examples

use tenferro_tensor_core::{Tensor, TensorRef};

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

Trait Implementations

impl<'a> Clone for TensorRef<'a>

fn clone(&self) -> TensorRef<'a>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<'a> Debug for TensorRef<'a>

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

Formats the value using the given formatter.