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 . |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/sparse/from_dense