TensorFlow函数：tf.sparse_retain

tf.sparse_retain 函数

sparse_retain(
    sp_input,
    to_retain
)

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

参见指南：稀疏张量>操纵

在一个 SparseTensor 中保留指定的非空值.

例如,如果 sp_input 有形状 [4, 5] 和4个非空字符串值,如下所示：

[0, 1]: a
[0, 3]: b
[2, 0]: c
[3, 1]: d

并且 to_retain = [True, False, False, True],则输出将是一个形状为 [4, 5] 以及具有2个非空值的 SparseTensor：

[0, 1]: a
[3, 1]: d

函数参数：

• sp_input：输入的 SparseTensor 带有 N 个非空元素.
• to_retain：长度为 N 的具有 M 个真值的 bool 向量.

函数返回值：

该函数返回一个与输入具有相同形状并且有 M 个非空元素的 SparseTensor,它对应于 to_retain 的真实位置.

可能引发的异常：

• TypeError：如果 sp_input 不是 SparseTensor.