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

• 節(jié)約內(nèi)存

• 在表達(dá)一個(gè)稀疏矩陣時(shí)候，不用一個(gè)很大的矩陣

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)>

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。