tf.io.serialize_sparse

View source on GitHub

Serialize a SparseTensor into a 3-vector (1-D Tensor) object.

Aliases:

• tf.compat.v1.io.serialize_sparse
• tf.compat.v1.serialize_sparse
• tf.serialize_sparse

tf.io.serialize_sparse(
    sp_input,
    name=None,
    out_type=tf.dtypes.string
)

Args:

• sp_input: The input SparseTensor.
• name: A name prefix for the returned tensors (optional).
• out_type: The dtype to use for serialization.

Returns:

A 3-vector (1-D Tensor), with each column representing the serialized SparseTensor's indices, values, and shape (respectively).

Raises:

• TypeError: If sp_input is not a SparseTensor.