experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add docstring to tf.sparse.from_dense.

Browse Source 
PiperOrigin-RevId: 254857335
This commit is contained in:
A. Unique TensorFlower 2019-06-24 16:09:57 -07:00 committed by TensorFlower Gardener
parent e7d3bf8ec8
commit 4f30dfe6bb
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tensorflow/python/ops/sparse_ops.py
View File

@ -102,6 +102,18 @@ def _make_int64_tensor(value, name):
@tf_export("sparse.from_dense") @tf_export("sparse.from_dense")
def from_dense(tensor, name=None): def from_dense(tensor, name=None):
"""Converts a dense tensor into a sparse tensor.
Only elements not equal to zero will be present in the result. The resulting
`SparseTensor` has the same dtype and shape as the input.
Args:
tensor: A dense `Tensor` to be converted to a `SparseTensor`.
name: Optional name for the op.
Returns:
The `SparseTensor`.
"""
with ops.name_scope(name, "dense_to_sparse"): with ops.name_scope(name, "dense_to_sparse"):
tensor = ops.convert_to_tensor(tensor) tensor = ops.convert_to_tensor(tensor)
indices = array_ops.where_v2( indices = array_ops.where_v2(