Trait tc_tensor::TensorReduce

pub trait TensorReduce<D: Dir> {
    type Txn: Transaction<D>;
    type Reduce: TensorInstance;
    fn product(self, axis: usize) -> TCResult<Self::Reduce>;
    fn product_all(&self, txn: Self::Txn) -> TCBoxTryFuture<'_, Number>;
    fn sum(self, axis: usize) -> TCResult<Self::Reduce>;
    fn sum_all(&self, txn: Self::Txn) -> TCBoxTryFuture<'_, Number>;
}

Tensor reduction operations

Associated Types

type Txn: Transaction<D>

The type of Transaction to expect

type Reduce: TensorInstance

The result type of a reduce operation

Required methods

fn product(self, axis: usize) -> TCResult<Self::Reduce>

Return the product of this Tensor along the given axis.

fn product_all(&self, txn: Self::Txn) -> TCBoxTryFuture<'_, Number>

Return the product of all elements in this Tensor.

fn sum(self, axis: usize) -> TCResult<Self::Reduce>

Return the sum of this Tensor along the given axis.

fn sum_all(&self, txn: Self::Txn) -> TCBoxTryFuture<'_, Number>

Return the sum of all elements in this Tensor.

Implementors

impl<FD, FS, D, T> TensorReduce<D> for Tensor<FD, FS, D, T> where
    D: Dir,
    T: Transaction<D>,
    FD: File<Array>,
    FS: File<Node>,
    D::File: AsType<FD> + AsType<FS>,
    D::FileClass: From<TensorType>, 

type Txn = T

type Reduce = Self

impl<FD, FS, D, T, A> TensorReduce<D> for SparseTensor<FD, FS, D, T, A> where
    D: Dir,
    T: Transaction<D>,
    FD: File<Array>,
    FS: File<Node>,
    D::File: AsType<FD> + AsType<FS>,
    D::FileClass: From<TensorType>,
    A: SparseAccess<FD, FS, D, T>,
    Self: TensorInstance,
    <Self as TensorInstance>::Dense: TensorReduce<D, Txn = T> + Send + Sync, 

type Txn = T

type Reduce = SparseTensor<FD, FS, D, T, SparseReduce<FD, FS, D, T>>

impl<FD, FS, D, T, B> TensorReduce<D> for DenseTensor<FD, FS, D, T, B> where
    D: Dir,
    T: Transaction<D>,
    FD: File<Array>,
    FS: File<Node>,
    D::File: AsType<FD> + AsType<FS>,
    B: DenseAccess<FD, FS, D, T>,
    D::FileClass: From<TensorType>, 

type Txn = T

type Reduce = DenseTensor<FD, FS, D, T, BlockListReduce<FD, FS, D, T, B>>