tf.where()【能懂版】

用法1：tf.where(x, y, z)

首先要求 x，y，z 的形状相同，其中 x 是条件矩阵（数据类型为 tf.bool 的 tensor），tf.where(x, y, z) 返回的结果 res 与 x，y，z 形状相同，而 res 的内容是：

• 若 x(i, j) 为 True，则 res(i, j) = y(i, j)
• 若 x(i, j) 为 False，则 res(i, j) = z(i, j)
  

用法2：tf.where(x)

当 y 和 z 都为 None 时得到这种情况，其中 x 是条件矩阵（数据类型为 tf.bool 的 tensor），tf.where(x) 返回 x 中为 True 的值的坐标矩阵