tensorflow--稀疏张量--3--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。