TensorFlow 1 version
|
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.
|