module

slicing

Ops that slice, index, stack, concat etc.

Functions

• ​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.