Trait tc_tensor::TensorIndex
source · [−]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;
}
Expand description
Tensor
indexing operations
Required Associated Types
type Txn: Transaction<D>
type Txn: Transaction<D>
The type of Transaction
to expect
type Index: TensorInstance
type Index: TensorInstance
The type of Tensor
returned by argmax
.
Required Methods
Return the indices of the maximum values in this Tensor
along the given axis
.