Skip to content

Commit 7a3dc23

Browse files
zhwesky2010Enigmatisms
authored andcommitted
some create api support more usage (PaddlePaddle#74494)
1 parent 3db16a2 commit 7a3dc23

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

python/paddle/utils/decorator_utils.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -123,6 +123,21 @@ def __init__(
123123
self.default_params = default_params
124124
warnings.simplefilter("always", category=Warning)
125125

126+
127+
# *size => shape decorator
128+
class SizeArgsDecorator(DecoratorBase):
129+
"""
130+
Usage Example:
131+
132+
paddle.ones(1, dtype=paddle.float32)
133+
paddle.ones(1, 2, 3, dtype=paddle.float32)
134+
paddle.ones([1, 2, 3], dtype=paddle.float32)
135+
paddle.ones(size=[1, 2, 3], dtype=paddle.float32)
136+
137+
paddle.ones([1, 2, 3], paddle.float32)
138+
paddle.ones(shape=[1, 2, 3], dtype=paddle.float32)
139+
"""
140+
126141
def process(
127142
self, args: tuple[Any, ...], kwargs: dict[str, Any]
128143
) -> tuple[tuple[Any, ...], dict[str, Any]]:

0 commit comments

Comments
 (0)