应用NPU兼容的dropout_with_byte_mask操作,仅支持NPU设备。此方法生成无状态随机uint8掩码,并根据该掩码执行dropout。
[object Object]
- input1 (
[object Object]): 必选参数,输入张量。 - p (
[object Object]):可选参数,dropout概率,默认值为0.5。 - training (
[object Object]):可选参数,是否启动dropout,当设置为True时启动,False时不启动。默认值为True。 - inplace (
[object Object]):可选参数,是否原地生效,当设置为True时将原地修改入参包含的值。默认值为False。
仅在设备32核场景下性能提升。
[object Object]