-
Notifications
You must be signed in to change notification settings - Fork 6.8k
Conversation
np_out = np.nonzero(x.asnumpy()) | ||
np_out = np.transpose(np_out) | ||
assert_almost_equal(mx_out.asnumpy(), np_out, rtol, atol) | ||
|
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.
One more blank line here.
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.
fcca31c
to
97cb5d6
Compare
97cb5d6
to
236ddcf
Compare
(1, 1), | ||
(1, 2, 3), | ||
(1, 0), | ||
(2, 0, 3) |
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.
Get rid of this shape
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.
for hybridize in [True, False]: | ||
for shape in [(), | ||
(1,), | ||
(1, 1), |
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.
Get rid of this shape and the shape above.
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.
aded5c8
to
c33b8b0
Compare
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 after rebase and resolution of conflicts
d10c07e
to
eb41507
Compare
* add cpu test and handle 0-dim * add FGradient with MakeZeroGradNodes * handle 0-dim and 0-shape and add test on gpu * add doc * fix bug in review * do not use thrust::inclusive_scan on cpu * fix format error * edit test and remove gpu test The output is same as numpy.transpose(numpy.nonzero(x)) * fix error of review * edit test
eb41507
to
d49dc5b
Compare
* add cpu test and handle 0-dim * add FGradient with MakeZeroGradNodes * handle 0-dim and 0-shape and add test on gpu * add doc * fix bug in review * do not use thrust::inclusive_scan on cpu * fix format error * edit test and remove gpu test The output is same as numpy.transpose(numpy.nonzero(x)) * fix error of review * edit test
* add cpu test and handle 0-dim * add FGradient with MakeZeroGradNodes * handle 0-dim and 0-shape and add test on gpu * add doc * fix bug in review * do not use thrust::inclusive_scan on cpu * fix format error * edit test and remove gpu test The output is same as numpy.transpose(numpy.nonzero(x)) * fix error of review * edit test
* add cpu test and handle 0-dim * add FGradient with MakeZeroGradNodes * handle 0-dim and 0-shape and add test on gpu * add doc * fix bug in review * do not use thrust::inclusive_scan on cpu * fix format error * edit test and remove gpu test The output is same as numpy.transpose(numpy.nonzero(x)) * fix error of review * edit test
numpy operator nonzero