🚀 Feature

Implement torch.where(condition) as an alias for a NumPy-like nonzero. Currently, we only support torch.where(condition, x, y) with the second and third arguments.

Motivation

NumPy supports numpy.where(condition) without the x and y arguments. It's behaves like numpy.nonzero.

https://docs.scipy.org/doc/numpy/reference/generated/numpy.where.html

Pitch

Maybe @alexpeys will stop bugging me if you implement this

Alternatives

Move @alexpeys's desk farther away