W3cubDocs

/TensorFlow 2.4

tf.compat.v1.sparse_split

Split a SparseTensor into num_split tensors along axis. (deprecated arguments)

If the sp_input.dense_shape[axis] is not an integer multiple of num_split each slice starting from 0:shape[axis] % num_split gets extra one dimension. For example, if axis = 1 and num_split = 2 and the input is:

input_tensor = shape = [2, 7]
[    a   d e  ]
[b c          ]

Graphically the output tensors are:

output_tensor[0] =
[    a   ]
[b c     ]

output_tensor[1] =
[ d e  ]
[      ]
Args
keyword_required Python 2 standin for * (temporary for argument reorder)
sp_input The SparseTensor to split.
num_split A Python integer. The number of ways to split.
axis A 0-D int32 Tensor. The dimension along which to split. Must be in range [-rank, rank), where rank is the number of dimensions in the input SparseTensor.
name A name for the operation (optional).
split_dim Deprecated old name for axis.
Returns
num_split SparseTensor objects resulting from splitting value.
Raises
TypeError If sp_input is not a SparseTensor.
ValueError If the deprecated split_dim and axis are both non None.

© 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/compat/v1/sparse_split