tf.ragged.boolean_mask

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

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

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительных сведений.

tf.compat.v1.ragged.boolean_mask, `tf.compat.v2.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)

<pre class="devsite-click-to-copy prettyprint lang-py">
  <code class="devsite-terminal" data-terminal-prefix="&gt;&gt;&gt;">tf.ragged.boolean_mask(  # Mask a 2D Tensor.</code>
  <code class="devsite-terminal" data-terminal-prefix="...">    data=[[1, 2, 3], [4, 5, 6], [7, 8, 9]],</code>
  <code class="devsite-terminal" data-terminal-prefix="...">    mask=[[T, F, T], [F, F, F], [T, F, F]]).tolist()</code>
  <code class="no-select nocode">  [[1, 3], [], [7]]</code>
  <code class="no-select nocode">  </code>
</pre>


<pre class="devsite-click-to-copy prettyprint lang-py">
  <code class="devsite-terminal" data-terminal-prefix="&gt;&gt;&gt;">tf.ragged.boolean_mask(  # Mask a 2D RaggedTensor.</code>
  <code class="devsite-terminal" data-terminal-prefix="...">    tf.ragged.constant([[1, 2, 3], [4], [5, 6]]),</code>
  <code class="devsite-terminal" data-terminal-prefix="...">    tf.ragged.constant([[F, F, T], [F], [T, T]])).tolist()</code>
  <code class="no-select nocode">  [[3], [], [5, 6]]</code>
  <code class="no-select nocode">  </code>
</pre>


<pre class="devsite-click-to-copy prettyprint lang-py">
  <code class="devsite-terminal" data-terminal-prefix="&gt;&gt;&gt;">tf.ragged.boolean_mask(  # Mask rows of a 2D RaggedTensor.</code>
  <code class="devsite-terminal" data-terminal-prefix="...">    tf.ragged.constant([[1, 2, 3], [4], [5, 6]]),</code>
  <code class="devsite-terminal" data-terminal-prefix="...">    tf.ragged.constant([True, False, True])).tolist()</code>
  <code class="no-select nocode">  [[1, 2, 3], [5, 6]]</code>
  <code class="no-select nocode">  </code>
</pre>

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