def dropout_with_byte_mask()

API接口

def 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核场景下性能提升。