#include <sparse_ops.h>
Split a SparseTensor
into num_split
tensors along one dimension.
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 ] [ ]
Arguments:
[0, rank(shape))
.Returns:
OutputList
output_indicesOutputList
output_values: A list of 1-D tensors represents the values of the output sparse tensors.OutputList
output_shape: A list of 1-D tensors represents the shape of the output sparse tensors. Constructors and Destructors | |
---|---|
SparseSplit(const ::tensorflow::Scope & scope, ::tensorflow::Input split_dim, ::tensorflow::Input indices, ::tensorflow::Input values, ::tensorflow::Input shape, int64 num_split) |
Public attributes | |
---|---|
output_indices | |
output_shape | |
output_values |
::tensorflow::OutputList output_indices
::tensorflow::OutputList output_shape
::tensorflow::OutputList output_values
SparseSplit( const ::tensorflow::Scope & scope, ::tensorflow::Input split_dim, ::tensorflow::Input indices, ::tensorflow::Input values, ::tensorflow::Input shape, int64 num_split )
© 2017 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/api_docs/cc/class/tensorflow/ops/sparse-split.html