此页面由 Cloud Translation API 翻译。
Switch to English

tf.sparse.retain

TensorFlow 1版 GitHub上查看源代码

保留了内指定的非空值SparseTensor

例如,如果sp_input具有形状[4, 5]和4个非空字符串值:

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

to_retain = [True, False, False, True] ,那么输出将是一个SparseTensor形状的[4, 5]用2非空值:

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

sp_input 输入SparseTensorN非空元素。
to_retain 的长度A布尔矢量NM真值。

SparseTensor与相同形状的输入和M对应于真实位置非空元素to_retain

TypeError 如果sp_input不是SparseTensor