TensorFlow函数:tf.sparse_retain
2018-03-07 10:31 更新
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.