Skip to content

Commit 8f5e51e

Browse files
authored
fix many type (#7423)
1 parent 098c765 commit 8f5e51e

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

66 files changed

+74
-74
lines changed

docs/api/paddle/Model_cn.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -263,7 +263,7 @@ summary(input_size=None, dtype=None)
263263
**参数**
264264

265265
- **input_size** (tuple|InputSpec|list[tuple|InputSpec],可选) - 输入 Tensor 的大小。如果网络只有一个输入,那么该值需要设定为 tuple 或 InputSpec。如果模型有多个输入。那么该值需要设定为 list[tuple|InputSpec],包含每个输入的 shape 。如果该值没有设置,会将 ``self._inputs`` 作为输入。默认值:None。
266-
- **dtype** (str,可选) - 输入 Tensor 的数据类型,如果没有给定,默认使用 ``float32`` 类型。默认值:None。
266+
- **dtype** (str|paddle.dtype|np.dtype,可选) - 输入 Tensor 的数据类型,如果没有给定,默认使用 ``float32`` 类型。默认值:None。
267267

268268
**返回**
269269

docs/api/paddle/Tensor_cn.rst

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -655,7 +655,7 @@ astype(dtype)
655655
将 Tensor 的类型转换为 ``dtype``,并返回一个新的 Tensor。
656656

657657
参数:
658-
- **dtype** (str) - 转换后的 dtype,支持'bool','float16','float32','float64','int8','int16',
658+
- **dtype** (str|paddle.dtype|np.dtype,可选) - 转换后的 dtype,支持'bool','float16','float32','float64','int8','int16',
659659
'int32','int64','uint8'。
660660

661661
返回:类型转换后的新的 Tensor
@@ -3431,7 +3431,7 @@ new_full(size, fill_value, \*, dtype=None, device=None, requires_grad=False, pin
34313431
- **fill_value** (Scalar|Tensor) - 用于填充的常量值。若为 Tensor,则应为标量(0 维 Tensor)。
34323432

34333433
关键字参数:
3434-
- **dtype** (str,可选) - 输出 Tensor 的数据类型,可选:``float16``、``float32``、``float64``、``int32``、``int64``、``complex64``、``complex128``。
3434+
- **dtype** (str|paddle.dtype|np.dtype,可选) - 输出 Tensor 的数据类型,可选:``float16``、``float32``、``float64``、``int32``、``int64``、``complex64``、``complex128``。
34353435
若为 None,则默认与 ``self`` 的 dtype 一致。
34363436
- **out** (Tensor,可选) - 用于保存输出结果的 Tensor,默认值为 None。
34373437
- **device** (PlaceLike|None,可选) - 期望创建 Tensor 所在的设备。若为 None,则与 ``self`` 保持一致。
@@ -3455,7 +3455,7 @@ new_ones(size, \*, dtype=None, device=None, requires_grad=False, pin_memory=Fals
34553455
若为列表或元组,其中元素需为整数或 0 维 Tensor。
34563456

34573457
关键字参数:
3458-
- **dtype** (str,可选) - 输出 Tensor 的数据类型,可选:``float16``、``float32``、``float64``、``int32``、``int64``、``complex64``、``complex128``。
3458+
- **dtype** (str|paddle.dtype|np.dtype,可选) - 输出 Tensor 的数据类型,可选:``float16``、``float32``、``float64``、``int32``、``int64``、``complex64``、``complex128``。
34593459
若为 None,则默认与 ``self`` 的 dtype 一致。
34603460
- **out** (Tensor,可选) - 用于保存输出结果的 Tensor,默认值为 None。
34613461
- **device** (PlaceLike|None,可选) - 期望创建 Tensor 所在的设备。若为 None,则与 ``self`` 保持一致。
@@ -3479,7 +3479,7 @@ new_zeros(size, \*, dtype=None, device=None, requires_grad=False, pin_memory=Fal
34793479
若为列表或元组,其中元素需为整数或 0 维 Tensor。
34803480

34813481
关键字参数:
3482-
- **dtype** (str,可选) - 输出 Tensor 的数据类型,可选:``float16``、``float32``、``float64``、``int32``、``int64``、``complex64``、``complex128``。
3482+
- **dtype** (str|paddle.dtype|np.dtype,可选) - 输出 Tensor 的数据类型,可选:``float16``、``float32``、``float64``、``int32``、``int64``、``complex64``、``complex128``。
34833483
若为 None,则默认与 ``self`` 的 dtype 一致。
34843484
- **out** (Tensor,可选) - 用于保存输出结果的 Tensor,默认值为 None。
34853485
- **device** (PlaceLike|None,可选) - 期望创建 Tensor 所在的设备。若为 None,则与 ``self`` 保持一致。
@@ -3503,7 +3503,7 @@ new_empty(size, \*, dtype=None, device=None, requires_grad=False, pin_memory=Fal
35033503
若为列表或元组,其中元素需为整数或 0 维 Tensor。
35043504

35053505
关键字参数:
3506-
- **dtype** (str,可选) - 输出 Tensor 的数据类型,可选:``float16``、``float32``、``float64``、``int32``、``int64``、``complex64``、``complex128``。
3506+
- **dtype** (str|paddle.dtype|np.dtype,可选) - 输出 Tensor 的数据类型,可选:``float16``、``float32``、``float64``、``int32``、``int64``、``complex64``、``complex128``。
35073507
若为 None,则默认与 ``self`` 的 dtype 一致。
35083508
- **out** (Tensor,可选) - 用于保存输出结果的 Tensor,默认值为 None。
35093509
- **device** (PlaceLike|None,可选) - 期望创建 Tensor 所在的设备。若为 None,则与 ``self`` 保持一致。

docs/api/paddle/amp/auto_cast_cn.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ auto_cast
1717
- **custom_white_list** (set|list,可选) - 飞桨有默认的白名单,通常不需要设置自定义白名单。自定义白名单中的算子在计算时会被认为是数值安全的,并且对性能至关重要。如果设置了该名单,其中的算子会使用 float16/bfloat16 计算。
1818
- **custom_black_list** (set|list,可选) - 飞桨有默认的黑名单,可以根据模型特点设置自定义黑名单。自定义黑名单中的算子在计算时会被认为是数值危险的,它们的影响也可能会在下游算子中观察到。该名单中的算子不会转为 float16/bfloat16 计算。
1919
- **level** (str,可选) - 混合精度训练的优化级别,可为 ``O1`` 、``O2`` 或者 ``OD`` 模式。在 O1 级别下,在白名单中的算子将使用 float16/bfloat16 计算,在黑名单中的算子将使用 float32 计算。在 O2 级别下,模型的参数需要调用 ``decorate`` 转换为 float16/bfloat16, 如果算子的浮点型输入全是 float16/bfloat16,算子才会采用 float16/bfloat16 计算,若任意浮点型输入是 float32 类型,算子将采用 float32 计算。在 OD 级别下,飞桨默认的白名单中的算子,如卷积和矩阵乘运算使用 float16/bfloat16 计算,其他算子均使用 float32 计算。默认为 O1。
20-
- **dtype** (str,可选) - 使用的数据类型,可以是 float16 或 bfloat16。默认为 float16。
20+
- **dtype** (str|paddle.dtype|np.dtype,可选) - 使用的数据类型,可以是 float16 或 bfloat16。默认为 float16。
2121
- **use_promote** (bool,可选) - 当一个算子存在 float32 类型的输入时,按照 Promote to the Widest 原则,选择 float32 数据类型进行计算。仅在 AMP-O2 训练时可配置。默认为 True。
2222

2323

docs/api/paddle/amp/decorate_cn.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ decorate
1717
- **models** (Layer|list of Layer) - 网络模型。在 ``O2`` 模式下,输入的模型参数将由 float32 转为 float16 或 bfloat16。
1818
- **optimizers** (Optimizer|list of Optimizer,可选) - 优化器,默认值为 None,若传入优化器或由优化器组成的 list 列表,将依据 master_weight 对优化器的 master_weight 属性进行设置。
1919
- **level** (str,可选) - 混合精度训练模式,默认 ``O1`` 模式。
20-
- **dtype** (str,可选) - 混合精度训练数据类型使用 float16 还是 bfloat16,默认为 float16 类型。
20+
- **dtype** (str|paddle.dtype|np.dtype,可选) - 混合精度训练数据类型使用 float16 还是 bfloat16,默认为 float16 类型。
2121
- **master_weight** (bool|None,可选) - 是否使用 master weight 策略。支持 master weight 策略的优化器包括 ``adam``、``adamW``、``momentum``,默认值为 None,在 ``O2`` 模式下使用 master weight 策略。
2222
- **save_dtype** (str|None,可选) - 网络存储类型,可为 float16、bfloat16、float32、float64。通过 ``save_dtype`` 可指定通过 ``paddle.save`` 和 ``paddle.jit.save`` 存储的网络参数数据类型。默认为 None,采用现有网络参数类型进行存储。
2323
- **master_grad** (bool, 可选) - 在 ``O2`` 模式下是否使用 float32 类型的权重梯度进行梯度裁剪、权重衰减、权重更新等计算。如果被启用,在反向传播结束后权重的梯度将会是 float32 类型。默认值:False,模型仅保存一份 float16 类型的权重梯度。

docs/api/paddle/arange_cn.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ arange
1515
- **start** (float|int|Tensor) - 区间起点(且区间包括此值)。当 ``start`` 类型是 Tensor 时,是形状为[]且数据类型为 int32、int64、float32、float64 的 0-D Tensor。如果仅指定 ``start``,而 ``end`` 为 None,则区间为[0, ``start``)。默认值为 0。
1616
- **end** (float|int|Tensor,可选) - 区间终点(且通常区间不包括此值)。当 ``end`` 类型是 Tensor 时,是形状为[]且数据类型为 int32、int64、float32、float64 的 0-D Tensor。默认值为 None。
1717
- **step** (float|int|Tensor,可选) - 均匀分割的步长。当 ``step`` 类型是 Tensor 时,是形状为[]且数据类型为 int32、int64、float32、float64 的 0-D Tensor。默认值为 1。
18-
- **dtype** (str|np.dtype,可选) - 输出 Tensor 的数据类型,支持 int32、int64、float32、float64。当该参数值为 None 时,输出 Tensor 的数据类型为 int64。默认值为 None。
18+
- **dtype** (str|paddle.dtype|np.dtype,可选) - 输出 Tensor 的数据类型,支持 int32、int64、float32、float64。当该参数值为 None 时,输出 Tensor 的数据类型为 int64。默认值为 None。
1919

2020
关键字参数
2121
::::::::::::

docs/api/paddle/argmax_cn.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ argmax
1313
- **x** (Tensor) - 输入的多维 ``Tensor``,支持的数据类型:float16、float32、float64、int16、int32、int64、uint8。
1414
- **axis** (int,可选) - 指定对输入 Tensor 进行运算的轴,``axis`` 的有效范围是[-R, R),R 是输入 ``x`` 的维度个数,``axis`` 为负数时,进行计算的 ``axis`` 与 ``axis`` + R 一致。默认值为 None,将会对输入的 `x` 进行平铺展开,返回最大值的索引。
1515
- **keepdim** (bool,可选) - 是否在输出 Tensor 中保留减小的维度。如果 keepdim 为 True,则输出 Tensor 和 x 具有相同的维度(减少的维度除外,减少的维度的大小为 1),默认值为 False。
16-
- **dtype** (np.dtype|str,可选) - 输出 Tensor 的数据类型,可选值为 int32,int64,默认值为 int64,将返回 int64 类型的结果。
16+
- **dtype** (str|paddle.dtype|np.dtype,可选) - 输出 Tensor 的数据类型,可选值为 int32,int64,默认值为 int64,将返回 int64 类型的结果。
1717
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。
1818

1919
返回

docs/api/paddle/argmin_cn.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ argmin
1313
- **x** (Tensor) - 输入的多维 ``Tensor``,支持的数据类型:float16、float32、float64、int16、int32、int64、uint8。
1414
- **axis** (int,可选) - 指定对输入 Tensor 进行运算的轴, ``axis`` 的有效范围是[-R, R),R 是输入 ``x`` 的维度个数, ``axis`` 为负数时,进行计算的 ``axis`` 与 ``axis`` + R 一致。默认值为 None,将会对输入的 `x` 进行平铺展开,返回最小值的索引。
1515
- **keepdim** (bool,可选) - 是否保留进行最小值索引操作的轴,默认值为 False。
16-
- **dtype** (np.dtype|str,可选) - 输出 Tensor 的数据类型,可选值为 int32、int64,默认值为'int64',将返回 int64 类型的结果。
16+
- **dtype** (str|paddle.dtype|np.dtype,可选) - 输出 Tensor 的数据类型,可选值为 int32、int64,默认值为'int64',将返回 int64 类型的结果。
1717
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。
1818

1919
返回

docs/api/paddle/audio/features/LogMelSpectrogram_cn.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ LogMelSpectrogram
2626
- **ref_value** (float,可选) - 参照值,如果小于 1.0,信号的 db 会被提升,相反 db 会下降,默认值为 1.0。
2727
- **amin** (float,可选) - 输入的幅值的最小值,默认是 1e-10。
2828
- **top_db** (float,可选) - log-mel 谱的最大值(db),默认是 None。
29-
- **dtype** (str,可选) - 输入和窗的数据类型,默认是``float32``。
29+
- **dtype** (str|paddle.dtype|np.dtype,可选) - 输入和窗的数据类型,默认是``float32``。
3030

3131

3232
返回

docs/api/paddle/audio/features/MFCC_cn.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ MFCC
2727
- **ref_value** (float,可选) - 参照值, 如果小于 1.0,信号的 db 会被提升, 相反 db 会下降, 默认值为 1.0。
2828
- **amin** (float,可选) - 输入的幅值的最小值,默认是 1e-10。
2929
- **top_db** (float,可选) - log-mel 谱的最大值(db),默认是 None。
30-
- **dtype** (str,可选) - 输入和窗的数据类型,默认是'float32'。
30+
- **dtype** (str|paddle.dtype|np.dtype,可选) - 输入和窗的数据类型,默认是'float32'。
3131

3232
返回
3333
:::::::::

docs/api/paddle/audio/features/MelSpectrogram_cn.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ MelSpectrogram
2323
- **f_max** (float,可选) - 最大频率(hz),默认为 None。
2424
- **htk** (bool,可选) - 在计算 fbank 矩阵时是否用在 HTK 公式缩放,默认是 False。
2525
- **norm** (Union[str, float],可选) -计算 fbank 矩阵时正则化的种类,默认是'slaney',也可以 norm=0.5,使用 p-norm 正则化。
26-
- **dtype** (str,可选) - 输入和窗的数据类型,默认是'float32'。
26+
- **dtype** (str|paddle.dtype|np.dtype,可选) - 输入和窗的数据类型,默认是'float32'。
2727

2828

2929
返回

0 commit comments

Comments
 (0)