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。

参数说明

约束说明

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

支持的型号