[object Object]

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

Description: Rolls data in a tensor based on the given size and dimension.

The following is an example:

[object Object]
[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]
  • Parameters:

    • x (aclTensor*, compute input): input original data, aclTensor on the device. The shape supports 0 to 8 dimensions. are supported. The can be ND.

      • [object Object]Atlas training products[object Object] and [object Object]Atlas inference products[object Object]: The data type can be FLOAT16, FLOAT32, INT8, UINT8, INT32, INT64, UINT32, or BOOL.
      • [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 BFLOAT16, FLOAT16, FLOAT32, INT8, UINT8, INT32, INT64, UINT32, or BOOL.
    • [object Object] (aclIntArray*, compute input): number of steps by which elements are rolled in each dimension. It is [object Object] on the host. The array length must be the same as that of [object Object].

    • [object Object] (aclIntArray*, compute input): dimensions along which elements are rolled. It is [object Object] on the host. The array length must be the same as that of [object Object]. The value range is [–x.dim(), x.dim() – 1]. For example, if x has 4 dimensions, the value range is [–4, 3].

    • [object Object] (aclTensor*, compute output): output data after rolling. It is [object Object] on the device. The shape supports 0–8 dimensions and must be the same as that of x. The data type must be the same as that of x. The supports ND.

      • [object Object]Atlas training products[object Object] and [object Object]Atlas inference products[object Object]: The data type can be FLOAT16, FLOAT32, INT8, UINT8, INT32, INT64, UINT32, or BOOL.
      • [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 BFLOAT16, FLOAT16, FLOAT32, INT8, UINT8, INT32, INT64, UINT32, or BOOL.
    • [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]
[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 the first-phase API aclnnRollGetWorkSpaceSize.

    • [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 compute:
    • [object Object] defaults to a deterministic implementation.
[object Object]

The following example is for reference only. For details, see .

[object Object]