FilesLast commitLast update
8 months ago
27 days ago
1 month ago
27 days ago
27 days ago
27 days ago
27 days ago
1 month ago
9 days ago
README

stateless_drop_out_gen_mask

English | 简体中文

Product Support Status

Product Supported
Ascend 950PR&950DT products √
Atlas A3 products ×
Atlas A2 products ×
Atlas 200I/500 A2 inference products ×
Atlas inference products ×
Atlas training products ×

Function Description

  • Operator function: Computes the random number input parameters key and counter based on seed, seed1, and offset. Then, it calls the philox_random algorithm to generate random numbers, calls the uniform algorithm to normalize the random numbers of the uint32 type, and finally calls the compare_scalar comparison function to output the mask result.

Parameter Description

Parameter Name Input/Output/Attribute Description Data Type Data Format
shape Input Obtains the size of the input shape. INT32, INT64 ND
prob Input Obtains the keep probability. FLOAT16, BF16, FLOAT ND
seed Input Obtains the random seed. INT32, INT64 ND
seed1 Input Obtains the random seed. INT32, INT64 ND
offset Input Step size for obtaining values. INT64 ND
out Output Obtains the output tensor. UINT8 ND

Constraint Description

None

Invocation Description

Invocation Method Sample Code Description
aclnn invocation test_aclnn_stateless_drop_out_gen_mask Invokes the stateless_drop_out_gen_mask operator through the aclnnDropoutGenMask API.