（beta）torch_npu.contrib.module.NpuCachedDropout

产品支持情况

产品	是否支持
Atlas A3 训练系列产品	√
Atlas A2 训练系列产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√

功能说明

在NPU设备上使用FairseqDropout。

函数原型

torch_npu.contrib.module.NpuCachedDropout(p, module_name=None)

参数说明

• p (float)：元素归零的概率。
• module_name (string)：模块名称。

约束说明

不支持动态shape。