Python tf.boolean_mask用法及代碼示例

將布爾掩碼應用於張量。

用法

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

Numpy 等價物是 tensor[mask] 。

通常，0 < dim(mask) = K <= dim(tensor) 和 mask 的形狀必須與 tensor 的形狀的前 K 個維度匹配。然後我們有： boolean_mask(tensor, mask)[i, j1,...,jd] = tensor[i1,...,iK,j1,...,jd] 其中 (i1,...,iK) 是 mask 的第 i 個 True 條目(行主要順序)。 axis 可以與 mask 一起使用來指示要屏蔽的軸。在這種情況下，axis + dim(mask) <= dim(tensor) 和 mask 的形狀必須與 tensor 的形狀的第一個 axis + dim(mask) 尺寸相匹配。

另請參閱：tf.ragged.boolean_mask，它可以應用於密集和參差不齊的張量，如果您需要保留 tensor 的掩碼尺寸(而不是像 tf.boolean_mask 那樣將它們展平)，可以使用它。

例子：

tensor = [0, 1, 2, 3]  # 1-D example
mask = np.array([True, False, True, False])
tf.boolean_mask(tensor, mask)
<tf.Tensor:shape=(2,), dtype=int32, numpy=array([0, 2], dtype=int32)>

tensor = [[1, 2], [3, 4], [5, 6]] # 2-D example
mask = np.array([True, False, True])
tf.boolean_mask(tensor, mask)
<tf.Tensor:shape=(2, 2), dtype=int32, numpy=
array([[1, 2],
       [5, 6]], dtype=int32)>

例子：

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