[object Object][object Object]

[object Object]undefined[object Object]

Operator function: Returns the product of each row in a specified dimension of the input tensor.

[object Object]

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

• [object Object]
• [object Object]

[object Object]

• Parameters:

  • [object Object] (aclTensor*, compute input): aclTensor on the device. are supported. The can be ND.

    • [object Object]Atlas inference products[object Object] and [object Object]Atlas training products[object Object]: The data type can be FLOAT16, FLOAT, DOUBLE, INT8, UINT8, INT16, INT32, INT64, BOOL, COMPLEX64, or COMPLEX128. If the input is an empty tensor, the output type cannot be COMPLEX64 or COMPLEX128.
    • [object Object]Atlas A2 training products/Atlas A2 inference products[object Object] and [object Object]Atlas A3 training products/Atlas A3 inference products[object Object]: The data type can be FLOAT16, FLOAT, DOUBLE, INT8, UINT8, INT16, INT32, INT64, BOOL, COMPLEX64, COMPLEX128, or BFLOAT16. If the input is an empty tensor, the output type cannot be COMPLEX64 or COMPLEX128.

  • [object Object](int64_t, compute input): int64_t on the host, the dimension to be reduced.

  • [object Object](bool, compute input): bool on the host, whether to reserve dimensions for the output tensor.

  • [object Object] (const aclDataType, compute input): aclDataType on the host, which specifies the data type of the output tensor. The data type must be the same as that of out.

  • [object Object] (aclTensor*, compute output): aclTensor on the device. The data type is the same as that of dtype. are supported. The can be ND.

    • [object Object]Atlas inference products[object Object] and [object Object]Atlas training products[object Object]: The data type can be FLOAT16, FLOAT, DOUBLE, INT8, UINT8, INT16, INT32, INT64, BOOL, COMPLEX64, or COMPLEX128.
    • [object Object]Atlas A2 training products/Atlas A2 inference products[object Object] and [object Object]Atlas A3 training products/Atlas A3 inference products[object Object]: The data type can be FLOAT16, FLOAT, DOUBLE, INT8, UINT8, INT16, INT32, INT64, BOOL, COMPLEX64, COMPLEX128, or BFLOAT16.

  • [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]

• Parameters:

  • [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 the first-phase API aclnnProdDimGetWorkspaceSize.

  • [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]

The following examples are for reference only. For details, see .