Skip to content

Commit da364ce

Browse files
zhwesky2010Enigmatisms
authored andcommitted
some create api support more usage (PaddlePaddle#74494)
1 parent bc441d4 commit da364ce

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
@@ -91,18 +91,19 @@ def process(
9191
return args, processed_kwargs
9292

9393

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

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

0 commit comments

Comments
 (0)