Argmax

Argmax#

MIOpen: Argmax
Argmax

Functions

miopenStatus_t miopenArgmaxForward (miopenHandle_t handle, const miopenTensorDescriptor_t xDesc, const void *x, const int32_t dim, const miopenTensorDescriptor_t yDesc, void *y)
 Find the index of the maximum value of a tensor across dimensions. To enable this, define MIOPEN_BETA_API before including miopen.h. More...
 

Detailed Description

Function Documentation

◆ miopenArgmaxForward()

miopenStatus_t miopenArgmaxForward ( miopenHandle_t  handle,
const miopenTensorDescriptor_t  xDesc,
const void *  x,
const int32_t  dim,
const miopenTensorDescriptor_t  yDesc,
void *  y 
)

Find the index of the maximum value of a tensor across dimensions. To enable this, define MIOPEN_BETA_API before including miopen.h.

Parameters
handleMIOpen handle (input)
xDescTensor descriptor for data input tensor x (input)
xData tensor x (input)
dimDimensions to reduce argmax. (input)
yDescTensor descriptor for output indice data tensor y (input)
yData tensor y (output)
Returns
miopenStatus_t
Examples
/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-miopen/checkouts/docs-6.1.1/include/miopen/miopen.h.