Skip to content

Conversation

zhwesky2010
Copy link
Contributor

@zhwesky2010 zhwesky2010 commented Mar 20, 2023

PR types

New features

PR changes

APIs

Describe

支持以下API输出0D Tensor:

1. paddle.argmin
2. paddle.argmax
3. paddle.median
4. paddle.kthvalue
5. paddle.mode
6. paddle.equal_all
7. paddle.allclose

Copy link
Contributor

@DDDivano DDDivano left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@zhwesky2010 zhwesky2010 merged commit cdefcd0 into PaddlePaddle:develop Mar 21, 2023
@zhwesky2010 zhwesky2010 changed the title [Zero-Dim] Support output 0D for argmin/argmax/median/kthvalue/mode/equal_all/allclose [Zero-Dim] Support output 0D for argmin/argmax/median/kthvalue/mode/equal_all/allclose, test=allcase Apr 11, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants