TensorFlow函数:tf.sparse_mask

由 Carrie 创建, 最后一次修改 2018-02-27

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实例.

以上内容是否对您有帮助:
二维码
建议反馈
二维码