Skip to content

Commit 6cf981b

Browse files
zhwesky2010Enigmatisms
authored andcommitted
some create api support more usage (PaddlePaddle#74494)
1 parent 788361a commit 6cf981b

File tree

1 file changed

+12
-11
lines changed

1 file changed

+12
-11
lines changed

python/paddle/utils/decorator_utils.py

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -92,18 +92,19 @@ def process(
9292
return args, processed_kwargs
9393

9494

95-
class ForbidKeywordsDecorator(DecoratorBase):
96-
"""A decorator that hints users to use the correct `compat` functions, when erroneous keyword arguments are detected"""
95+
# *size => shape decorator
96+
class SizeArgsDecorator(DecoratorBase):
97+
"""
98+
Usage Example:
9799
98-
def __init__(
99-
self, illegal_keys: list[str], func_name: str, correct_name: str
100-
) -> None:
101-
super().__init__()
102-
self.illegal_keys = (
103-
[illegal_keys] if isinstance(illegal_keys, str) else illegal_keys
104-
)
105-
self.func_name = func_name
106-
self.correct_name = correct_name
100+
paddle.ones(1, dtype=paddle.float32)
101+
paddle.ones(1, 2, 3, dtype=paddle.float32)
102+
paddle.ones([1, 2, 3], dtype=paddle.float32)
103+
paddle.ones(size=[1, 2, 3], dtype=paddle.float32)
104+
105+
paddle.ones([1, 2, 3], paddle.float32)
106+
paddle.ones(shape=[1, 2, 3], dtype=paddle.float32)
107+
"""
107108

108109
def process(
109110
self, args: tuple[Any, ...], kwargs: dict[str, Any]

0 commit comments

Comments
 (0)