[代码]tensorflow代码片段-稀疏矩阵

稀疏的string tensor 转dense, 每行一个值

tf.sparse.to_dense(sp_x)

或者

def to_dense_mx1(sp):
    sp_x = tf.SparseTensor(indices=tf.concat((sp.indices[:,0:1], tf.zeros_like(sp.indices[:,0:1])), 1),
                           values=sp.values, dense_shape=tf.stack((sp.dense_shape[0], 1)))
    return tf.sparse.to_dense(sp_x)
posted @ 2022-01-21 17:51  bregman  阅读(79)  评论(0)    收藏  举报