Enum TypedTensorWrite 

pub enum TypedTensorWrite<'a, T> {
    Tensor(&'a mut TypedTensor<T>),
    View(TypedTensorViewMut<'a, T>),
}

Mutable typed tensor output accepted by synchronous eager kernels.

TypedTensorWrite is the typed counterpart to TensorWrite. It accepts either an owned compact TypedTensor or an arbitrary-strided mutable TypedTensorViewMut without erasing the scalar type at the public API boundary.

Examples

use tenferro_tensor::{TypedTensorViewMut, TypedTensorWrite};

let mut data = [0.0_f64, 1.0, 0.0, 2.0];
let view = TypedTensorViewMut::from_slice([2], [2], 1, &mut data)?;
let write = TypedTensorWrite::from_view(view).into_tensor_write();
assert_eq!(write.shape(), &[2]);
assert_eq!(write.strides()?, [2]);

Variants

Tensor(&'a mut TypedTensor<T>)

An owned compact typed tensor borrowed mutably for the write.

View(TypedTensorViewMut<'a, T>)

An arbitrary-strided mutable typed tensor view.

Implementations

impl<'a, T> TypedTensorWrite<'a, T>

pub fn from_tensor(tensor: &'a mut TypedTensor<T>) -> Self

Create a writable target from an owned typed tensor.

Examples

use tenferro_tensor::{TypedTensor, TypedTensorWrite};

let mut tensor = TypedTensor::<f64>::from_vec_col_major(vec![1], vec![0.0])?;
let write = TypedTensorWrite::from_tensor(&mut tensor);
assert!(matches!(write, TypedTensorWrite::Tensor(_)));

pub fn from_view(view: TypedTensorViewMut<'a, T>) -> Self

Create a writable target from a mutable typed tensor view.

Examples

use tenferro_tensor::{TypedTensorViewMut, TypedTensorWrite};

let mut data = [0.0_f64, 1.0];
let view = TypedTensorViewMut::from_col_major(&[2], &mut data)?;
let write = TypedTensorWrite::from_view(view);
assert!(matches!(write, TypedTensorWrite::View(_)));

impl<'a, T: TensorScalar> TypedTensorWrite<'a, T>

pub fn into_tensor_write(self) -> TensorWrite<'a>

Erase the scalar type while preserving the output layout.

Examples

use tenferro_tensor::{DType, TypedTensor, TypedTensorWrite};

let mut tensor = TypedTensor::<f64>::from_vec_col_major(vec![2], vec![0.0; 2])?;
let write = TypedTensorWrite::from_tensor(&mut tensor).into_tensor_write();
assert_eq!(write.dtype(), DType::F64);
assert_eq!(write.shape(), &[2]);

Trait Implementations

impl<'a, T: Debug> Debug for TypedTensorWrite<'a, T>

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

Formats the value using the given formatter.

impl<'a, T> From<&'a mut TypedTensor<T>> for TypedTensorWrite<'a, T>

fn from(tensor: &'a mut TypedTensor<T>) -> Self

Converts to this type from the input type.

impl<'a, T> From<TypedTensorViewMut<'a, T>> for TypedTensorWrite<'a, T>

fn from(view: TypedTensorViewMut<'a, T>) -> Self

Converts to this type from the input type.