We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3532dd8 commit 1aa91bfCopy full SHA for 1aa91bf
python/paddle/tensor/attribute.py
@@ -104,9 +104,9 @@ def shape(input: Tensor) -> Tensor:
104
105
>>> img = np.ones((3, 100, 100)).astype(np.float32)
106
107
- >>> res = exe.run(paddle.static.default_main_program(), feed={'x':img}, fetch_list=[output])
+ >>> res = exe.run(paddle.static.default_main_program(), feed={'x': img}, fetch_list=[output])
108
>>> print(res)
109
- [array([ 3, 100, 100], dtype=int32)]
+ [array([ 3, 100, 100], dtype=int64)]
110
"""
111
if in_dynamic_or_pir_mode():
112
out = _C_ops.shape64(input) # type: ignore
0 commit comments