Tensor.where should return condition.nonzero() when condition is the only argument

The linked issue is already closed! But the place to discuss these is on github itself. It looks like your PR has some changes requested, so you can address those in by leaving a comment on the issue if you’re confused/stuck, or by making the requested changes.