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

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


在 SparseTensor 中保留指定的非空值。

用法

tf.sparse.retain(
    sp_input, to_retain
)

参数

  • sp_input 输入 SparseTensor 和 N 非空元素。
  • to_retain 一个长度为 N 且具有 M 真值的布尔向量。

返回

  • 与输入具有相同形状的 SparseTensor 和与 to_retain 中的真实位置相对应的 M 非空元素。

抛出

  • TypeError 如果 sp_input 不是 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] 的 SparseTensor ，具有 2 个非空值：

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

相关用法


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