当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.sparse.mask用法及代码示例


屏蔽 IndexedSlices 的元素。

用法

tf.sparse.mask(
    a, mask_indices, name=None
)

参数

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

返回

  • 被屏蔽的 IndexedSlices 实例。

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

当您需要提取 IndexedSlices 对象中的切片子集时,这很有用。

例如:

# `a` contains slices at indices [12, 26, 37, 45] from a large tensor
# with shape [1000, 10]
a.indices  # [12, 26, 37, 45]
tf.shape(a.values)  # [4, 10]

# `b` will be the subset of `a` slices at its second and third indices, so
# we want to mask its first and last indices (which are at absolute
# indices 12, 45)
b = tf.sparse.mask(a, [12, 45])

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

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.sparse.mask。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。