View source on GitHub |
Converts a dense tensor into a sparse tensor.
tf.sparse.from_dense(
tensor, name=None
)
Only elements not equal to zero will be present in the result. The resulting
SparseTensor
has the same dtype and shape as the input.
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]])
Args | |
---|---|
tensor
|
A dense Tensor to be converted to a SparseTensor .
|
name
|
Optional name for the op. |
Returns | |
---|---|
The SparseTensor .
|