tf.boolean_mask

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

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

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

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

tf.compat.v1.boolean_mask

tf.boolean_mask(
    tensor, mask, name='boolean_mask', axis=None
)

Эквивалент в NumPy — tensor[mask].

# 1-D example
tensor = [0, 1, 2, 3]
mask = np.array([True, False, True, False])
boolean_mask(tensor, mask)  # [0, 2]

В общем случае, 0 < dim(mask) = K <= dim(tensor), и mask's форма должны соответствовать первым K размерностям формы tensor. Тогда у нас есть: boolean_mask(tensor, mask)[i, j1,...,jd] = tensor[i1,...,iK,j1,...,jd] где (i1,...,iK) — i-й True элемент mask (порядок строк). axis можно использовать с mask для указания оси, с которой происходит маскирование. В этом случае, axis + dim(mask) <= dim(tensor) и mask's форма должны соответствовать первым axis + dim(mask) размерностям формы tensor.

См. также: tf.ragged.boolean_mask, который может быть применён к плотным и разрозненным тензорам и может быть использован, если вам нужно сохранить маскируемые размерности tensor (а не сплющивать их, как делает tf.boolean_mask).

Аргументы
`tensor`	N-мерный тензор.
`mask`	K-мерный булевый тензор, K <= N и K должно быть статически известно.
`name`	Имя для этой операции (необязательно).
`axis`	0-мерный целочисленный тензор, представляющий ось в `tensor` для маскирования. По умолчанию ось равна 0, что означает маскирование с первой размерности. В противном случае K + ось <= N.

Возвращает
(N-K+1)-мерный тензор, заполненный элементами в `tensor`, соответствующими значениям `True` в `mask`.

Исключения
`ValueError`	Если формы не соответствуют.

Примеры:

# 2-D example
tensor = [[1, 2], [3, 4], [5, 6]]
mask = np.array([True, False, True])
boolean_mask(tensor, mask)  # [[1, 2], [5, 6]]

© 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/boolean_mask