-
Notifications
You must be signed in to change notification settings - Fork 2.3k
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
【Hackathon 5th No.97】add paddle tanh_shrink op #20067
Conversation
@yuxu42 Could you help to have a review on this PR, thanks |
@xczhai Could you please help review? Thanks! |
data = np.random.uniform(-10, 20, [4, 3, 2]).astype(np.float32) | ||
tanh_shrink("tanh_shrink_2", data) | ||
|
||
# data = np.random.uniform(10, 20, [2, 3, 4]).astype(np.float16) // only on gpu |
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.
please remove the useless codes if unnecessary.
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.
done
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
@Asthestarsfalll please rebase the PR. triger the CI again. Thanks. |
@ceciliapeng2011 Hi Cecilia, could you help merge the PR? Thanks. |
@Asthestarsfalll could you rebase and fix the CI issue? Thanks! |
Details:
add paddle tanh_shrink op
Tickets:
waitting for
Reference:
Docs: https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/nn/functional/tanhshrink_cn.html#tanhshrink
Support data type:
UnitTest: https://github.com/PaddlePaddle/Paddle/blob/29059a349810a7cb895d84b90f1b177fed952791/test/legacy_test/test_activation_op.py#L1094
Code: https://github.com/PaddlePaddle/Paddle/blob/release/2.5/python/paddle/nn/functional/activation.py#L1404
PaddlePaddle/Paddle#57262
Test
waiting for: