torch.where的新用法（很老但是大家忽略的用法）

torch.where(condition) → tuple of LongTensor

torch.where(condition) is identical to torch.nonzero(condition, as_tuple=True).

通常 ，我们都 会这样使用torch.where函数:

torch.where(condition, x, y)

但是实际上torch.where还有如题头如示的用法，返回tuple.

tuple中是condition中符合条件的值的index.

比如

a=[1,2,3,4,5]

torch.where（a>3)

返回( 3,4)