為什么用tf.sparse.SparseTensor的原因

• 節約內存

• 在表達一個稀疏矩陣時候，不用一個很大的矩陣

tf.sparse.SparseTensor(indices, values, dense_shape )

• indices ：非零值的元素的索引

• values ：非零值

• dense_shape ：shape

import tensorflow as tf sp_input=tf.sparse.SparseTensor(indices=[[0, 0], [1, 2]], values=[1, 3], dense_shape=[3, 4])tf.sparse.to_dense(sp_input, default_value=None, validate_indices=True, name=None ) <tf.Tensor: shape=(3, 4), dtype=int32, numpy= array([[1, 0, 0, 0],[0, 0, 3, 0],[0, 0, 0, 0]], dtype=int32)>

總結

以上是生活随笔為你收集整理的tf.sparse.SparseTensor的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。