TensorFlow函数：tf.sparse_mask

tf.sparse_mask 函数

sparse_mask(
    a,
    mask_indices,
    name=None
)

定义在：tensorflow/python/ops/array_ops.py.

请参阅指南：变量>稀疏变量更新

IndexedSlices 要屏蔽(mask)的元素.

给定一个 IndexedSlices 实例 a,返回另一个包含 a 切片的子集的 IndexedSlices.只返回未在 mask_indices 中指定的索引中的切片.

这在您需要在 IndexedSlices 对象中提取切片的子集时非常有用.

例如：

# `a` 包含在索引 [12, 26, 37, 45] 的切片,它来自一个带有形状 [1000, 10] 的大张量a.indices  # [12, 26, 37, 45]
tf.shape(a.values)  # [4, 10]

# `b`将是 `a` 切片在其第二和第三个指数中的子集, 因此我们要掩盖其第一个和最后一个索引 (这是绝对索引 12, 45)
b = tf.sparse_mask(a, [12, 45])

b.indices  # [26, 37]
tf.shape(b.values)  # [2, 10]

函数参数：

• a：一个IndexedSlices实例.
• mask_indices：要屏蔽的元素的索引.
• name：操作的名称(可选).

函数返回值：

被屏蔽的IndexedSlices实例.