昇腾社区首页
中文
注册
开发者
下载

(beta)torch_npu.contrib.module.NpuCachedDropout

产品支持情况

[object Object]undefined

功能说明

在NPU设备上使用FairseqDropout。

函数原型

[object Object]

参数说明

  • p ([object Object]):元素归零的概率。
  • module_name ([object Object]):模块名称。

约束说明

不支持动态shape。