Trait tc_tensor::TensorIndex

pub trait TensorIndex<D: Dir> {
    type Txn: Transaction<D>;
    type Index: TensorInstance;

    fn argmax<'async_trait>(
        self,
        txn: Self::Txn,
        axis: usize
    ) -> Pin<Box<dyn Future<Output = TCResult<Self::Index>> + Send + 'async_trait>>
   where
        Self: 'async_trait;
    fn argmax_all<'async_trait>(
        self,
        txn: Self::Txn
    ) -> Pin<Box<dyn Future<Output = TCResult<u64>> + Send + 'async_trait>>
   where
        Self: 'async_trait;
}

Tensor indexing operations

Required Associated Types

type Txn: Transaction<D>

The type of Transaction to expect

type Index: TensorInstance

The type of Tensor returned by argmax.

Required Methods

fn argmax<'async_trait>(
    self,
    txn: Self::Txn,
    axis: usize
) -> Pin<Box<dyn Future<Output = TCResult<Self::Index>> + Send + 'async_trait>>where
    Self: 'async_trait,

Return the indices of the maximum values in this Tensor along the given axis.

fn argmax_all<'async_trait>(
    self,
    txn: Self::Txn
) -> Pin<Box<dyn Future<Output = TCResult<u64>> + Send + 'async_trait>>where
    Self: 'async_trait,

Return the offset of the maximum value in this Tensor.

Implementors

impl<FD, FS, D, T> TensorIndex<D> for Tensor<FD, FS, D, T>where
    FD: File<Array>,
    FS: File<Node>,
    D: Dir,
    T: Transaction<D>,
    <D::Read as DirRead>::FileEntry: AsType<FD> + AsType<FS>,
    <D::Write as DirWrite>::FileClass: From<BTreeType> + From<TensorType>,

type Txn = T

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

impl<FD, FS, D, T, A> TensorIndex<D> for SparseTensor<FD, FS, D, T, A>where
    FD: File<Array>,
    FS: File<Node>,
    D: Dir,
    T: Transaction<D>,
    A: SparseWrite<FD, FS, D, T>,
    <D::Read as DirRead>::FileEntry: AsType<FD> + AsType<FS>,
    <D::Write as DirWrite>::FileClass: From<BTreeType> + From<TensorType>,

type Txn = T

type Index = SparseTensor<FD, FS, D, T, SparseTable<FD, FS, D, T>>

impl<FD, FS, D, T, B> TensorIndex<D> for DenseTensor<FD, FS, D, T, B>where
    D: Dir,
    T: Transaction<D>,
    FD: File<Array>,
    FS: File<Node>,
    <D::Read as DirRead>::FileEntry: AsType<FD> + AsType<FS>,
    B: DenseWrite<FD, FS, D, T>,
    <D::Write as DirWrite>::FileClass: From<BTreeType> + From<TensorType>,

type Txn = T

type Index = DenseTensor<FD, FS, D, T, BlockListFile<FD, FS, D, T>>