`tensor.where` does not behave like `np.where` when only one input (condition) is passed

[Ricardo Vieira]

Description

In this case it should return condition.nonzero(): https://numpy.org/doc/stable/reference/generated/numpy.where.html