（beta）torch_npu.contrib.function.dropout_with_byte_mask

接口原型

torch_npu.contrib.function.dropout_with_byte_mask(input1, p=0.5, training=True, inplace=False)

功能描述

应用NPU兼容的dropout_with_byte_mask操作，仅支持npu设备。这个dropout_with_byte_mask方法生成无状态随机uint8掩码，并根据掩码做dropout。

参数说明

• p：probability of a channel to be zeroed. Default: 0.5
• training：apply dropout if is ``True``. Default: ``True``
• inplace：If set to ``True``, will do this operation in-place. Default: ``False``

约束说明

仅在设备32核场景下性能提升。

支持的型号

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