tf.raw_ops.TakeManySparseFromTensorsMap

Read SparseTensors from a SparseTensorsMap and concatenate them.

The input sparse_handles must be an int64 matrix of shape [N, 1] where N is the minibatch size and the rows correspond to the output handles of AddSparseToTensorsMap or AddManySparseToTensorsMap. The ranks of the original SparseTensor objects that went into the given input ops must all match. When the final SparseTensor is created, it has rank one higher than the ranks of the incoming SparseTensor objects (they have been concatenated along a new row dimension on the left).

The output SparseTensor object's shape values for all dimensions but the first are the max across the input SparseTensor objects' shape values for the corresponding dimensions. Its first shape value is N, the minibatch size.

The input SparseTensor objects' indices are assumed ordered in standard lexicographic order. If this is not the case, after this step run SparseReorder to restore index ordering.

For example, if the handles represent an input, which is a [2, 3] matrix representing two original SparseTensor objects:

index = [ 0]
        [10]
        [20]
values = [1, 2, 3]
shape = [50]

and

index = [ 2]
        [10]
values = [4, 5]
shape = [30]

then the final SparseTensor will be:

index = [0  0]
        [0 10]
        [0 20]
        [1  2]
        [1 10]
values = [1, 2, 3, 4, 5]
shape = [2 50]
Args
sparse_handles A Tensor of type int64. 1-D, The N serialized SparseTensor objects. Shape: [N].
dtype A tf.DType. The dtype of the SparseTensor objects stored in the SparseTensorsMap.
container An optional string. Defaults to "". The container name for the SparseTensorsMap read by this op.
shared_name An optional string. Defaults to "". The shared name for the SparseTensorsMap read by this op. It should not be blank; rather the shared_name or unique Operation name of the Op that created the original SparseTensorsMap should be used.
name A name for the operation (optional).
Returns
A tuple of Tensor objects (sparse_indices, sparse_values, sparse_shape).
sparse_indices A Tensor of type int64.
sparse_values A Tensor of type dtype.
sparse_shape A Tensor of type int64.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/TakeManySparseFromTensorsMap