ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf.io.serialize_many_sparse

TensorFlow 2 version View source on GitHub

Serialize N-minibatch SparseTensor into an [N, 3] Tensor.

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 serialized SparseTensor objects going into each row of the output Tensor will have rank R-1.

The minibatch size N is extracted from sparse_shape[0].

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

A matrix (2-D Tensor) with N rows and 3 columns. Each column represents serialized SparseTensor's indices, values, and shape (respectively).

TypeError If sp_input is not a SparseTensor.