TensorFlow函數(shù)：tf.sparse_mask

tf.sparse_mask 函數(shù)

sparse_mask(
    a,
    mask_indices,
    name=None
)

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

請參閱指南：變量>稀疏變量更新

IndexedSlices 要屏蔽(mask)的元素.

給定一個 IndexedSlices 實(shí)例 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` 切片在其第二和第三個指數(shù)中的子集, 因此我們要掩蓋其第一個和最后一個索引 (這是絕對索引 12, 45)
b = tf.sparse_mask(a, [12, 45])

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

函數(shù)參數(shù)：

• a：一個IndexedSlices實(shí)例.
• mask_indices：要屏蔽的元素的索引.
• name：操作的名稱(可選).

函數(shù)返回值：

被屏蔽的IndexedSlices實(shí)例.