@@ -9,23 +9,23 @@ fused_dropout_add
99
1010参数
1111:::::::::
12- - **x ** (Tensor): 输入张量。数据类型为 bfloat16, float16, float32 或 float64.
13- - **y ** (Tensor): 输入张量,数据类型为 bfloat16, float16, float32 或 float64.
14- - **p ** (float|int, 可选): 将单位设置为零的概率。默认值: 0.5。
15- - **training ** (bool, 可选): 标记是否为训练阶段。默认值: True。
16- - **mode ** (str, 可选): ['upscale_in_train'(默认) | 'downscale_in_infer'].
12+ - **x ** (Tensor) - 输入张量。数据类型为 bfloat16、 float16、 float32 或 float64。
13+ - **y ** (Tensor) - 输入张量,数据类型为 bfloat16、 float16、 float32 或 float64。
14+ - **p ** (float|int, 可选) - 将单位设置为零的概率。默认值: 0.5。
15+ - **training ** (bool, 可选) - 标记是否为训练阶段。默认值: True。
16+ - **mode ** (str, 可选) - ['upscale_in_train'(默认) | 'downscale_in_infer']。
1717
18- 1. upscale_in_train (默认), 在训练时放大输出
18+ 1. upscale_in_train(默认), 在训练时放大输出
1919
2020 - 训练: :math: `out = x \times \frac {mask}{(1.0 - dropout\_prob)} + y`
2121 - 推理: :math: `out = x + y`
2222
23- 2. downscale_in_infer, 在推理时缩小输出
23+ 2. downscale_in_infer, 在推理时缩小输出
2424
2525 - 训练: :math: `out = input \times mask + y`
2626 - 推理: :math: `out = input \times (1.0 - dropout\_prob) + y`
2727
28- - **name ** (str, 可选): 操作的名称, 默认值: None。有关更多信息, 请参阅 :ref: `api_guide_Name `.
28+ - **name ** (str, 可选) - 操作的名称, 默认值: None。有关更多信息, 请参阅 :ref: `api_guide_Name `。
2929
3030
3131返回
0 commit comments