Skip to content

Commit 085801e

Browse files
zhwesky2010Enigmatisms
authored andcommitted
some create api support more usage (PaddlePaddle#74494)
1 parent cd8d6ae commit 085801e

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
@@ -127,6 +127,21 @@ def __init__(
127127
self.default_params = default_params
128128
warnings.simplefilter("always", category=Warning)
129129

130+
131+
# *size => shape decorator
132+
class SizeArgsDecorator(DecoratorBase):
133+
"""
134+
Usage Example:
135+
136+
paddle.ones(1, dtype=paddle.float32)
137+
paddle.ones(1, 2, 3, dtype=paddle.float32)
138+
paddle.ones([1, 2, 3], dtype=paddle.float32)
139+
paddle.ones(size=[1, 2, 3], dtype=paddle.float32)
140+
141+
paddle.ones([1, 2, 3], paddle.float32)
142+
paddle.ones(shape=[1, 2, 3], dtype=paddle.float32)
143+
"""
144+
130145
def process(
131146
self, args: tuple[Any, ...], kwargs: dict[str, Any]
132147
) -> tuple[tuple[Any, ...], dict[str, Any]]:

0 commit comments

Comments
 (0)