-
Notifications
You must be signed in to change notification settings - Fork 26.3k
C++ API: torch::nn::Softmin #27459
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
C++ API: torch::nn::Softmin #27459
Conversation
|
Thanks a lot for the great work! We will need some fixes as suggested in #27446. |
| struct TORCH_API SoftminOptions { | ||
| SoftminOptions(int64_t dim); | ||
|
|
||
| // Dimension along which Softmin will be computed. |
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.
nit: we should change // to /// so that the comment will show up in the C++ API docs (my apologies for missing it for the Softmax module 😞)
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.
If you don't mind, I'll also fix Softmax module in this PR.
What do you think about this idea?
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.
Yes that would be awesome, thanks so much for your help @nuka137!
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.
Fixed them.
|
I fixed the comments on |
yf225
left a comment
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.
@nuka137 The changes look awesome, thanks so much for your help! I will fix the merge conflicts :D
facebook-github-bot
left a comment
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.
@yf225 is landing this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
Summary: Add torch::nn::Softmin module and functional support for the C++ API. Related Issue: pytorch#25883 Reviewer: yf225 Pull Request resolved: pytorch#27459 Differential Revision: D17892852 Pulled By: yf225 fbshipit-source-id: db15b06e8ad33947e7d65995df700f5e90c3b6a8
Add torch::nn::Softmin module and functional support for the C++ API.
Related Issue: #25883
Reviewer: @yf225