为什么用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)>