Mojo function

split

split[n: Int](input: Symbol, sizes: IndexList[n], axis: Int = 0) -> List[Symbol]

Splits a symbolic tensor into specified bucket sizes along the axis.

Parameters:

• ​n (Int): The number of symbolic tensors to split into.

Args:

• ​input (Symbol): The symbolic tensor to split.
• ​sizes (IndexList[n]): The list of sizes for each split.
• ​axis (Int): The axis to split along.

Returns:

n symbolic tensor values. The ith result corresponds to the sizes[i]. Each tensor will have the same rank as the input, and will be the result of slicing the input of sizes[i] elements along axis, starting at the offset of the cumulative sum of the previous sizes.

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