Argmax

Functions

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

handle	MIOpen handle (input)
xDesc	Tensor descriptor for data input tensor x (input)
x	Data tensor x (input)
dim	Dimensions to reduce argmax. (input)
yDesc	Tensor descriptor for output indice data tensor y (input)
y	Data 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.