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。

个人公众号,比较懒,很少更新,可以在上面提问题,如果回复不及时,可发邮件给我: tiehan@sina.cn

Sam avatar
About Sam
专注生物信息 专注转化医学