Trait tc_tensor::TensorReduce

pub trait TensorReduce<D: Dir> {
    type Txn: Transaction<D>;
    type Reduce: TensorInstance;

    fn max(self, axis: usize, keepdims: bool) -> TCResult<Self::Reduce>;
    fn max_all(&self, txn: Self::Txn) -> TCBoxTryFuture<'_, Number>;
    fn min(self, axis: usize, keepdims: bool) -> TCResult<Self::Reduce>;
    fn min_all(&self, txn: Self::Txn) -> TCBoxTryFuture<'_, Number>;
    fn product(self, axis: usize, keepdims: bool) -> TCResult<Self::Reduce>;
    fn product_all(&self, txn: Self::Txn) -> TCBoxTryFuture<'_, Number>;
    fn sum(self, axis: usize, keepdims: bool) -> TCResult<Self::Reduce>;
    fn sum_all(&self, txn: Self::Txn) -> TCBoxTryFuture<'_, Number>;
}

Tensor reduction operations

Required 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 max(self, axis: usize, keepdims: bool) -> TCResult<Self::Reduce>

Return the maximum of this Tensor along the given axis.

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

Return the maximum element in this Tensor.

fn min(self, axis: usize, keepdims: bool) -> TCResult<Self::Reduce>

Return the minimum of this Tensor along the given axis.

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

Return the minimum element in this Tensor.

fn product(self, axis: usize, keepdims: bool) -> 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, keepdims: bool) -> 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
    FD: File<Key = u64, Block = Array, Inner = D::Inner>,
    FS: File<Key = NodeId, Block = Node>,
    D: Dir,
    T: Transaction<D>,
    D::Write: DirCreateFile<FD>,

type Txn = T

type Reduce = Tensor<FD, FS, D, T>

impl<FD, FS, D, T, A> TensorReduce<D> for SparseTensor<FD, FS, D, T, A>where
    FD: File<Key = u64, Block = Array, Inner = D::Inner>,
    FS: File<Key = NodeId, Block = Node>,
    D: Dir,
    T: Transaction<D>,
    A: SparseAccess<FD, FS, D, T>,
    D::Write: DirCreateFile<FD>,
    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
    FD: File<Key = u64, Block = Array, Inner = D::Inner>,
    FS: File<Key = NodeId, Block = Node>,
    D: Dir,
    T: Transaction<D>,
    B: DenseAccess<FD, FS, D, T>,
    D::Write: DirCreateFile<FD>,

type Txn = T

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