當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。