TensorFlow函數(shù)：tf.sparse_retain

tf.sparse_retain 函數(shù)

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

函數(shù)參數(shù)：

• sp_input：輸入的 SparseTensor 帶有 N 個非空元素.
• to_retain：長度為 N 的具有 M 個真值的 bool 向量.

函數(shù)返回值：

該函數(shù)返回一個與輸入具有相同形狀并且有 M 個非空元素的 SparseTensor,它對應(yīng)于 to_retain 的真實位置.

可能引發(fā)的異常：

• TypeError：如果 sp_input 不是 SparseTensor.