Skip to main content



Ops that slice, index, stack, concat etc.


  • gather: Selects elements out of an input tensor by index.
  • slice: Slices a symbolic tensor along each dimension.
  • split: Splits a symbolic tensor into specified bucket sizes along the axis.
  • concat: Concatenates a list of symbolic tensors along an axis.
  • stack: Stacks a list of tensors along a new axis.