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]]