tf.io.serialize_many_sparse | TensorFlow v2.16.1 (original) (raw)
tf.io.serialize_many_sparse
Stay organized with collections Save and categorize content based on your preferences.
Serialize N
-minibatch SparseTensor
into an [N, 3]
Tensor
.
tf.io.serialize_many_sparse(
sp_input,
out_type=tf.dtypes.string,
name=None
)
The SparseTensor
must have rank R
greater than 1, and the first dimension is treated as the minibatch dimension. Elements of the SparseTensor
must be sorted in increasing order of this first dimension. The serializedSparseTensor
objects going into each row of the output Tensor
will have rank R-1
.
The minibatch size N
is extracted from sparse_shape[0]
.
Args | |
---|---|
sp_input | The input rank R SparseTensor. |
out_type | The dtype to use for serialization. |
name | A name prefix for the returned tensors (optional). |
Returns |
---|
A matrix (2-D Tensor) with N rows and 3 columns. Each column represents serialized SparseTensor's indices, values, and shape (respectively). |
Raises | |
---|---|
TypeError | If sp_input is not a SparseTensor. |
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-04-26 UTC.