class NpuCachedDropout()

class NpuCachedDropout(torch.nn.Dropout):

在NPU设备上使用FairseqDropout。

• 参数：
  • p (Float) - 元素归零的概率。
  • module_name (String) - 模型名称。
• 调用方式样例：

  from torch_npu.contrib.module import NpuDropPath
  NpuCachedDropout.enable_dropout_ensemble(model)

• 示例：

    >>> model = NpuMNIST().to("npu")
    >>> x = torch.randn(2,10,16,16).to("npu")
    >>> NpuCachedDropout.enable_dropout_ensemble(model)
    >>> output = model(x)