Description: Computes the size of each kernel based on the input outputSize, performs 3D max pooling on the input self, and outputs the pooled value outputOut and index indicesOut. The difference between aclnnAdaptiveMaxPool3d and aclnnMaxPool3d is that you only need to specify the outputSize size and divide the pooling region based on the outputSize size.
Formula: N (Batch) represents the batch size, C (Channels) represents the number of feature map channels, D (Depth) represents the feature map depth, H (Height) represents the feature map height, and W (Width) represents the feature map width. For the input self with shape and outputSize with shape , the output has shape and the indices have shape . The calculation formula for each element in the corresponding tensor is as follows:
Each operator has calls. First, aclnnAdaptiveMaxPool3dGetWorkspaceSize is called to obtain the workspace size required for computation and the executor that contains the operator computation process. Then, aclnnAdaptiveMaxPool3d is called to perform computation.
Parameters:
[object Object]- For the [object Object]Atlas A3 training products/Atlas A3 inference products[object Object] and [object Object]Atlas A2 training products/Atlas A2 inference products[object Object], the data type of the
[object Object]parameter does not support INT64.
- For the [object Object]Atlas A3 training products/Atlas A3 inference products[object Object] and [object Object]Atlas A2 training products/Atlas A2 inference products[object Object], the data type of the
Returns:
[object Object]: status code. For details, see .The first-phase API implements input parameter verification. The following errors may be thrown.
[object Object]
Deterministic compute:
- aclnnAdaptiveMaxPool3d defaults to a deterministic implementation.
Shape description:
- self.shape = (N, C, Din, Hin, Win)
- outputSize = [Dout, Hout, Wout]
- outputOut.shape = (N, C, Dout, Hout, Wout)
- indicesOut.shape = (N, C, Dout, Hout, Wout)
The following example is for reference only. For details, see .