[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): input tensor, indicating the probability distribution of each object. It is the 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. 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 non-negative 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] (aclTensor*, computation input): aclTensor on the device. The shape is [1]. are supported. The supports ND, and the data type can be INT64. It can be used to set the seed value of the random number generator, which affects the generated random number sequence.
  • [object Object] (aclTensor*, computation input): aclTensor on the device. The shape is [1]. are supported. The supports ND, and the data type can be INT64. The sum of this value and the scalar offset is used as the offset of the random number operator. It indicates the offset of the random number, 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] (int64_t, computation input): integer on the host, which is used as the accumulation of offsetTensor.
  • [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 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 .