W3cubDocs

/TensorFlow C++

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 [-rank(value), rank(value)).

Returns:

  • OutputList: Tensors whose shape matches that of value except along axis, where their sizes are size_splits[i].
Constructors and Destructors
SplitV(const ::tensorflow::Scope & scope, ::tensorflow::Input value, ::tensorflow::Input size_splits, ::tensorflow::Input axis, int64 num_split)
Public attributes
output
Public functions
operator[](size_t index) const

Public attributes

output

::tensorflow::OutputList output

Public functions

SplitV

 SplitV(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input value,
  ::tensorflow::Input size_splits,
  ::tensorflow::Input axis,
  int64 num_split
)

operator[]

::tensorflow::Output operator[](
  size_t index
) const 

© 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/split-v.html