Mojo module

slicing

Ops that slice, index, stack, concat etc.

Functions

• ​concat: Concatenates a list of symbolic tensors along an axis.
• ​gather: Selects elements out of an input tensor by index.
• ​select: Returns condition ? x : y (element-wise), where cond, x and y are input tensors.
• ​slice: Slices a symbolic tensor along each dimension.
• ​split: Splits a symbolic tensor into specified bucket sizes along the axis.
• ​stack: Stacks a list of tensors along a new axis.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit