{ }
Split a SparseTensor
into num_split
tensors along one dimension.
tf.raw_ops.SparseSplit(
split_dim, indices, values, shape, num_split, name=None
)
If the shape[split_dim]
is not an integer multiple of num_split
. Slices
[0 : shape[split_dim] % num_split]
gets one extra dimension.
For example, if split_dim = 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] = shape = [2, 4]
[ a ]
[b c ]
output_tensor[1] = shape = [2, 3]
[ d e ]
[ ]