-
Notifications
You must be signed in to change notification settings - Fork 5.6k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[Zero-Dim] Support output 0D for squeeze, unbind, unstack. #52843
[Zero-Dim] Support output 0D for squeeze, unbind, unstack. #52843
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
x1 = paddle.full([1], 0) | ||
x2 = paddle.full([2], 2) | ||
|
||
[out1] = paddle.unstack(x1, 0) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这个API有反向,需要测一下反向。x1.grad/x2.grad
x1 = paddle.full([1], 0) | ||
x2 = paddle.full([2], 2) | ||
|
||
[out1] = paddle.unbind(x1, 0) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这个API有反向,需要测一下反向。x1.grad/x2.grad
@@ -4073,6 +4122,36 @@ def test_upsample(self): | |||
self.assertEqual(res1[0].shape, (2, 3, 12, 12)) | |||
self.assertEqual(res1[1].shape, (2, 3, 6, 6)) | |||
|
|||
@prog_scope() | |||
def test_unstack(self): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
测试case同静态图
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
你的PR已合入Paddle库,请关注后续测试结果。 |
PR types
New features
PR changes
APIs
Description
pcard-66984
支持API输出0D Tensor:squeeze, unbind, unstack.