torch_npu.contrib.NpuFairseqDropout
torch_npu.contrib.NpuFairseqDropout(p, module_name=None)
在npu设备上使用FairseqDropout。
参数解释:
p (float): probability of an element to be zeroed.
module_name (string): the name of the model.
约束条件:
不支持动态shape。
示例:
无
父主题:
torch_npu.contrib