tf.ragged.boolean_mask

Применяет булеву маску к data без сведения измерений маски в одно.

Просмотреть псевдонимы

Совместимые псевдонимы для миграции

Дополнительные сведения см. в руководстве по миграции.

tf.compat.v1.ragged.boolean_mask

tf.ragged.boolean_mask(
    data, mask, name=None
)

Используется в блокнотах

Используется в руководстве
Токенизаторы подслов

Возвращает потенциально нерегулярный тензор, сформированный из элементов data, для которых соответствующее значение в mask равно True.

output[a1...aA, i, b1...bB] = data[a1...aA, j, b1...bB]

Где j — это i-я запись True в mask[a1...aA].

Обратите внимание, что output сохраняет измерения маски a1...aA; это отличается от tf.boolean_mask, которая сводит эти измерения в одно.

Аргументы
`data`	Потенциально нерегулярный тензор.
`mask`	Потенциально нерегулярный булев тензор. Форма `mask` должна быть префиксом формы `data`. `rank(mask)` должна быть известна статически.
`name`	Префикс имени возвращаемого тензора (необязательно).

Возвращает
Потенциально нерегулярный тензор, сформированный из элементов `data`, для которых соответствующее значение в `mask` равно `True`. `rank(output) = rank(data)`. `output.ragged_rank = max(data.ragged_rank, rank(mask) - 1)`.

Вызывает исключение
`ValueError`	если `rank(mask)` не известно статически или если `mask.shape` не является префиксом `data.shape`.

Примеры:

# Aliases for True & False so data and mask line up.
T, F = (True, False)

tf.ragged.boolean_mask(  # Mask a 2D Tensor.
    data=[[1, 2, 3], [4, 5, 6], [7, 8, 9]],
    mask=[[T, F, T], [F, F, F], [T, F, F]]).to_list()
[[1, 3], [], [7]]

tf.ragged.boolean_mask(  # Mask a 2D RaggedTensor.
    tf.ragged.constant([[1, 2, 3], [4], [5, 6]]),
    tf.ragged.constant([[F, F, T], [F], [T, T]])).to_list()
[[3], [], [5, 6]]

tf.ragged.boolean_mask(  # Mask rows of a 2D RaggedTensor.
    tf.ragged.constant([[1, 2, 3], [4], [5, 6]]),
    tf.ragged.constant([True, False, True])).to_list()
[[1, 2, 3], [5, 6]]

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/ragged/boolean_mask