Returns the median value and position of the specified dimension of a tensor. If the number of elements in the specified dimension is size, the subscript of the median value after sorting is [object Object].
Each operator has calls. First, aclnnMedianDimGetWorkspaceSize is called to obtain the workspace size required for computation and the executor that contains the operator computation process. Then, aclnnMedianDim is called to perform computation.
Parameters
[object Object]- The [object Object]Atlas inference products[object Object] and [object Object]Atlas training products[object Object] do not support BFLOAT16.
Returns
[object Object]: status code. For details, see .The first-phase API performs input parameter validation. The following errors may be returned:
[object Object]
Deterministic compute:
- aclnnMedianDim defaults to a deterministic implementation.
If the data type of self is not FLOAT, FLOAT16, or BFLOAT16, the operator execution may time out due to an overlarge tensor size (an AI CPU error is reported, with reason=[aicpu timeout]). The maximum size for the specific type (closely related to the remaining memory of the machine) is as follows:
- INT64 type: 150000000
- UINT8, INT8, INT16, INT32 types: 725000000
When the data type of self is BFLOAT16, the case where the dim axis is 1 or scalar inputs are not supported.
The following example is for reference only. For details, see .