tensorflow::ops::Split

#include <array_ops.h>

Splits a tensor into num_split tensors along one dimension.

Summary

Arguments:

  • scope: A Scope object
  • axis: 0-D. The dimension along which to split. Must be in the range [-rank(value), rank(value)).
  • value: The tensor to split.
  • num_split: The number of ways to split. Must evenly divide value.shape[split_dim].

Returns:

  • OutputList: They are identically shaped tensors, whose shape matches that of value except along axis, where their sizes are values.shape[split_dim] / num_split.

Constructors and Destructors

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

Public attributes

output

Public functions

operator[](size_t index) const

Public attributes

output

::tensorflow::OutputList output

Public functions

Split

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

operator[]

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