-
Notifications
You must be signed in to change notification settings - Fork 6.8k
[MXNET-235] add axis support and gradient for L2norm #9740
Conversation
also need ord |
@piiswrong how many ord should we support? We can support ord=1, inf and None. |
Hi, the community has passed to vote about associating the code changes with JIRA (https://lists.apache.org/thread.html/ab22cf0e35f1bce2c3bf3bec2bc5b85a9583a3fe7fd56ba1bbade55f@%3Cdev.mxnet.apache.org%3E) We have updated the guidelines for contributors in https://cwiki.apache.org/confluence/display/MXNET/Development+Process, please ensure that you have created a JIRA at https://issues.apache.org/jira/projects/MXNET/issues/ to describe your work in this pull request and include the JIRA title in your PR as [MXNET-xxxx] your title where MXNET-xxxx is the JIRA id Thanks! |
@zheng-da ping |
I almost forgot this one. I'll look into it tomorrow. |
e0554ab
to
32a7f06
Compare
MSHADOW_XINLINE static DType Map(DType a) { | ||
return a * a; | ||
} | ||
}; |
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.
I think square is supported in mshadow_op::square
.
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.
needs a JIRA
@cjolivier01 I created a JIRA issue |
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.
thanks!
* Add axis support for norm. * Add tests on norm in diff cases. * Fix a compilation error. * Fix a lint complain. * Fix a compilation error. * Implement norm backward. * Fix a bug on computing norm on entire sparse arrays. * Fix tests. * Fix a bug in L2Norm on dense arrays. * Fix a lint error. * Remove square. * Add ord in Norm. * Fix a compile error. * Use NormParam. * Fix tests. * Update broadcast_reduce_op_value.cc
* Add axis support for norm. * Add tests on norm in diff cases. * Fix a compilation error. * Fix a lint complain. * Fix a compilation error. * Implement norm backward. * Fix a bug on computing norm on entire sparse arrays. * Fix tests. * Fix a bug in L2Norm on dense arrays. * Fix a lint error. * Remove square. * Add ord in Norm. * Fix a compile error. * Use NormParam. * Fix tests. * Update broadcast_reduce_op_value.cc
Description
This PR adds axis support and compute gradient for L2norm.
Checklist
Essentials
make lint
)Changes