Python tf.raw_ops.Select用法及代码示例

从 x 或 y 中选择元素，具体取决于 condition 。

用法

tf.raw_ops.Select(
    condition, x, y, name=None
)

x 和 y 张量必须具有相同的形状，并且输出也将具有该形状。

如果 x 和 y 是标量，则 condition 张量必须是标量。如果 x 和 y 是向量或更高等级，则 condition 必须是标量，大小与 x 的第一个维度匹配的向量，或者必须具有与 x 相同的形状。

condition 张量充当掩码，根据每个元素的值选择输出中的相应元素/行是否应取自 x(如果为真)或 y(如果为假)。

如果 condition 是一个向量并且 x 和 y 是更高等级的矩阵，那么它选择从 x 和 y 复制哪一行(外部维度)。如果 condition 与 x 和 y 具有相同的形状，则它选择从 x 和 y 复制哪个元素。

例如：

# 'condition' tensor is [[True,  False]
#                        [False, True]]
# 't' is [[1, 2],
#         [3, 4]]
# 'e' is [[5, 6],
#         [7, 8]]
select(condition, t, e)  # => [[1, 6], [7, 4]]


# 'condition' tensor is [True, False]
# 't' is [[1, 2],
#         [3, 4]]
# 'e' is [[5, 6],
#         [7, 8]]
select(condition, t, e) ==> [[1, 2],
                             [7, 8]]