（beta）at_npu::native::npu_dropout_gen_mask

定义文件

third_party\op-plugin\op_plugin\include\ops.h

接口原型

at::Tensor npu_dropout_gen_mask(const at::Tensor &self, at::IntArrayRef size, double p, int64_t seed, int64_t offset, c10::optional<bool> parallel, c10::optional<bool> sync)

功能描述

训练过程中，按照概率p随机生成mask，用于元素置零。

参数说明

• self ：Tensor类型，输入的张量。
• size：IntArrayRef类型，获取tensor的size。
• p：double类型，元素置0的概率。
• seed ：int64_t类型，随机数的种子，影响生成的随机数序列。
• offset ：int64_t类型，随机数的偏移量，影响生成随机数序列的位置。
• parallel ：可选参数bool类型，是否并行计算。
• sync：可选参数bool类型，是否做同步。

支持的型号

• Atlas 训练系列产品
• Atlas A2 训练系列产品
• Atlas 推理系列产品