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核场景下性能提升。
- 示例:
无