开发者
资源
[object Object][object Object][object Object]undefined
[object Object]

应用NPU兼容的dropout_with_byte_mask操作,仅支持NPU设备。此方法生成无状态随机uint8掩码,并根据该掩码执行dropout。

[object Object]
[object Object]
[object Object]
  • input1 ([object Object]): 必选参数,输入张量。
  • p ([object Object]):可选参数,dropout概率,默认值为0.5。
  • training ([object Object]):可选参数,是否启动dropout,当设置为True时启动,False时不启动。默认值为True。
  • inplace ([object Object]):可选参数,是否原地生效,当设置为True时将原地修改入参包含的值。默认值为False。
[object Object]

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

[object Object]
[object Object]