pub struct ArgMaxOperation<'c> { /* private fields */ }
Expand description
An argmax
operation. Perform argmax on the input..
This returns the index with the largest value across the given axis of the input tensor.
Implementations§
source§impl<'c> ArgMaxOperation<'c>
impl<'c> ArgMaxOperation<'c>
sourcepub fn as_operation(&self) -> &Operation<'c>
pub fn as_operation(&self) -> &Operation<'c>
Returns a generic operation.
sourcepub fn builder(
context: &'c Context,
location: Location<'c>
) -> ArgMaxOperationBuilder<'c, Unset, Unset, Unset>
pub fn builder( context: &'c Context, location: Location<'c> ) -> ArgMaxOperationBuilder<'c, Unset, Unset, Unset>
Creates a builder.
pub fn output(&self) -> Result<OperationResult<'c, '_>, Error>
pub fn input(&self) -> Result<Value<'c, '_>, Error>
pub fn axis(&self) -> Result<IntegerAttribute<'c>, Error>
pub fn set_axis(&mut self, value: IntegerAttribute<'c>)
Trait Implementations§
source§impl<'c> From<ArgMaxOperation<'c>> for Operation<'c>
impl<'c> From<ArgMaxOperation<'c>> for Operation<'c>
source§fn from(operation: ArgMaxOperation<'c>) -> Self
fn from(operation: ArgMaxOperation<'c>) -> Self
Converts to this type from the input type.
Auto Trait Implementations§
impl<'c> RefUnwindSafe for ArgMaxOperation<'c>
impl<'c> !Send for ArgMaxOperation<'c>
impl<'c> !Sync for ArgMaxOperation<'c>
impl<'c> Unpin for ArgMaxOperation<'c>
impl<'c> UnwindSafe for ArgMaxOperation<'c>
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more