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


Python tf.sparse.from_dense用法及代碼示例


將密集張量轉換為稀疏張量。

用法

tf.sparse.from_dense(
    tensor, name=None
)

參數

  • tensor 將要轉換為 SparseTensor 的密集 Tensor
  • name 操作的可選名稱。

返回

  • SparseTensor

結果中隻會出現不等於 0 的元素。生成的 SparseTensor 具有與輸入相同的 dtype 和 shape。

sp = tf.sparse.from_dense([0, 0, 3, 0, 1])
sp.shape.as_list()
[5]
sp.values.numpy()
array([3, 1], dtype=int32)
sp.indices.numpy()
array([[2],
       [4]])

相關用法


注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.sparse.from_dense。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。