tensorflow::ops::SplitV

tensorflow::ops::SplitV

#include <array_ops.h>

Splits a tensor into num_split tensors along one dimension.

Summary

Arguments:

• scope: A Scope object
• value: The tensor to split.
• size_splits: list containing the sizes of each output tensor along the split dimension. Must sum to the dimension of value along split_dim. Can contain one -1 indicating that dimension is to be inferred.
• axis: 0-D. The dimension along which to split. Must be in the range [0, rank(value)).

Returns:

• OutputList: Tensors whose shape matches that of value except along axis, where their sizes are size_splits[i].