[object Object][object Object][object Object]undefined[object Object]

Description: Computes the mean value of tensors on a specified dimension.

[object Object]

Each operator has calls. First, [object Object] is called to obtain the workspace size required for computation and the executor that contains the operator computation process. Then, [object Object] is called to perform computation.

• [object Object]
• [object Object]

[object Object]

• Parameter description:

  • [object Object] (aclTensor*, compute input): aclTensor on the device. If the input is an empty tensor, the output type cannot be COMPLEX64 or COMPLEX128. are supported. The can be ND.
    • [object Object]Atlas A2 training products/Atlas A2 inference products[object Object] and [object Object]Atlas A3 training series products/Atlas A3 inference series products[object Object]: The data type can be FLOAT, FLOAT16, BFLOAT16, DOUBLE, INT32, INT64, INT16, INT8, UINT8, COMPLEX64, or COMPLEX128.
    • [object Object]Atlas inference products[object Object] and [object Object]Atlas training products[object Object]: The data type can be FLOAT, FLOAT16, DOUBLE, INT32, INT64, INT16, INT8, UINT8, COMPLEX64, or COMPLEX128.
  • dim (aclIntArray*, input): aclIntArray on the host. The data type can be INT64. The value range is [–r, r – 1], where [object Object] is the dimension count of the input data.
  • [object Object] (bool, input): whether to keep the reduced dimension. The data type is BOOL.
  • [object Object] (bool, input): behavior defined when dim is set to []. If this parameter is set to [object Object], all axes are reduced when [object Object] is set to []. If this parameter is set to [object Object], all axes are retained when [object Object] is set to [], and the output tensor is the same as the input tensor.
  • [object Object] (aclTensor*, compute output): aclTensor on the device. The data type must be the same as that of [object Object]. are supported. The can be ND.
    • [object Object]Atlas A2 training products/Atlas A2 inference products[object Object] and [object Object]Atlas A3 training series products/Atlas A3 inference series products[object Object]: The data type can be FLOAT, FLOAT16, BFLOAT16, DOUBLE, INT32, INT64, INT16, INT8, UINT8, COMPLEX64, or COMPLEX128.
    • [object Object]Atlas inference products[object Object] and [object Object]Atlas training products[object Object]: The data type can be FLOAT, FLOAT16, DOUBLE, INT32, INT64, INT16, INT8, UINT8, COMPLEX64, or COMPLEX128.
  • [object Object] (uint64_t*, output): size of the workspace to be allocated on the device.
  • [object Object] (aclOpExecutor**, output): operator executor, containing the operator computation process.

• Returns:

  [object Object]: status code. For details, see .

[object Object]

• Parameter description:

  • [object Object] (void *, input): address of the workspace to be allocated on the device.
  • [object Object] (uint64_t, input): size of the workspace to be allocated on the device, which is obtained by calling [object Object].
  • [object Object] (aclOpExecutor *, input): operator executor, containing the operator computation process.
  • [object Object] (aclrtStream, input): stream for executing the task.

• Returns:

  [object Object]: status code. For details, see .

[object Object]

• Deterministic computation:
  • [object Object] defaults to a deterministic implementation.

[object Object]

For details, see .