where

该OP计算输入元素中为True的元素在输入中的坐标（index）。

• condition （Variable）– 输入秩至少为1的多维Tensor，数据类型是bool类型。

返回：输出condition元素为True的坐标（index），将所有的坐标（index）组成一个2-D的Tensor。

代码示例：