[Typing][A-27,A-28,A-32,A-33,A-35,A-36] Add type annotations for paddle/nn/initializer/*
#65206
Conversation
…addle/nn/initializer/*`
|
你的PR提交成功，感谢你对开源项目的贡献! |
| @@ -23,6 +28,9 @@ | |||
| ) | |||
| from .initializer import Initializer | |||
|
|
|||
| if TYPE_CHECKING: | |||
| import numpy | |||
There was a problem hiding this comment.
| import numpy | |
| import numpy.typing as npt |
| @@ -38,19 +46,21 @@ class NumpyArrayInitializer(Initializer): | |||
|
|
|||
| """ | |||
|
|
|||
| def __init__(self, value): | |||
| def __init__(self, value: numpy.ndarray): | |||
There was a problem hiding this comment.
| def __init__(self, value: numpy.ndarray): | |
| def __init__(self, value: npt.NDArray[Any]): |
| def __init__(self, value, name=None): | ||
| def __init__( | ||
| self, | ||
| value: numpy.ndarray | list[int] | tuple[int] | paddle.Tensor, |
There was a problem hiding this comment.
tuple[int] 表示 (1, ) 这种只有一个元素的，如果是不定长元素，应该用 tuple[int, ...]，在支持 list[T] | tuple[T, ...] 的情况下，更常见的是直接写 Sequence[T]
| fan_in: float | None = None, | ||
| seed: int = 0, | ||
| negative_slope: float = 0, | ||
| nonlinearity: str = 'relu', |
There was a problem hiding this comment.
是否直接用 Literal？见
Paddle/python/paddle/nn/initializer/initializer.py
Lines 167 to 180 in 8261cde
使用 TypeAlias 写在 initializer.py 里这边 import 过来就好
| def forward(self, var, block=None): | ||
| def forward( | ||
| self, var: paddle.Tensor, block: paddle.pir.Block | None = None | ||
| ) -> paddle.Tensor | None: |
| @@ -28,6 +29,22 @@ | |||
| ) | |||
| from .lazy_init import lazy_init_helper | |||
|
|
|||
| if TYPE_CHECKING: | |||
| NonLinearityT: TypeAlias = Literal[ | |||
There was a problem hiding this comment.
| NonLinearityT: TypeAlias = Literal[ | |
| _NonLinearity: TypeAlias = Literal[ |
这样吧，T 还是统一表示泛型参数吧
| @@ -16,6 +16,7 @@ | |||
|
|
|||
| import functools | |||
| import math | |||
| from typing import TYPE_CHECKING, Literal, TypeAlias | |||
There was a problem hiding this comment.
TypeAlias 要从 typing_extensions import
SigureMo
changed the title
paddle/nn/initializer/*paddle/nn/initializer/*
…dle/nn/initializer/*` (PaddlePaddle#65206)
PR Category
User Experience
PR Types
Improvements
Description
类型标注：
Related links
@SigureMo @megemini