[object Object]

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

Operator function: Draws numsamples samples from the input tensor based on the probability distribution of each object, and stores the indices of these samples in the output tensor.

[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:

  • [object Object] (aclTensor*, computation input): aclTensor on the device. The shape is (N, C) or (C). The value of [object Object] must be greater than or equal to 0, and the dimensions of [object Object] and [object Object] must be the same. are supported. The supports ND.

    • [object Object]Atlas inference products[object Object] and [object Object]Atlas training products[object Object]: The data type can be FLOAT16, FLOAT, or DOUBLE.
    • [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, FLOAT, or DOUBLE.

  • [object Object] (int64_t, computation input): integer on the host, indicating the number of samples to draw from each multinomial distribution. The value of [object Object] is a positive integer. When [object Object] is false, [object Object] is not greater than C.

  • [object Object] (bool, computation input): boolean type on the host, indicating whether to draw with replacement or not

  • [object Object] (int64_t, computation input): seed of the random number generator, which affects the generated random number sequence.

  • [object Object] (int64_t, computation input): offset of the random number generator, which affects the position of the generated random number sequence. After the offset is set, the generated random number sequence starts from the specified position.

  • [object Object] (aclTensor*, computation output): aclTensor on the device. The data type can be INT64. The shape is (N, numsamples) or (numsamples). The dimensions of [object Object] and [object Object] are the same. are supported. The supports ND.

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

[object Object]

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