Method ArgMax

ArgMax(TensorFloat, TensorInt, int, bool, bool)

Computes the indices of the maximum elements of the input tensor along a given axis.

Declaration

void ArgMax(TensorFloat X, TensorInt O, int axis, bool keepdim, bool selectLastIndex)

Parameters

Type	Name	Description
TensorFloat	X	The input tensor.
TensorInt	O	The output tensor to be computed and filled.
int	axis	The axis along which to reduce.
bool	keepdim	Whether to keep the reduced axes in the output tensor.
bool	selectLastIndex	Whether to perform the operation from the back of the axis.

ArgMax(TensorInt, TensorInt, int, bool, bool)

Computes the indices of the maximum elements of the input tensor along a given axis.

Declaration

void ArgMax(TensorInt X, TensorInt O, int axis, bool keepdim, bool selectLastIndex)

Parameters

Type	Name	Description
TensorInt	X	The input tensor.
TensorInt	O	The output tensor to be computed and filled.
int	axis	The axis along which to reduce.
bool	keepdim	Whether to keep the reduced axes in the output tensor.
bool	selectLastIndex	Whether to perform the operation from the back of the axis.