tf.sparse.from_dense

Преобразует плотный тензор в разреженный.

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

Только элементы, не равные нулю, будут присутствовать в результате. Результирующий SparseTensor имеет тот же тип и форму, что и входной.

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