For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).

Mojo function

extend_shape

def extend_shape[rank: Int](in_shape: IndexList[rank], first: Int, last: Int) -> IndexList[(rank + Int(2))]

Extend input shape by inserting first and last at both ends.

Parameters:

rank (Int): Spatial rank of the input shape.

Args:

in_shape (IndexList[rank]): Input shape to extend.
first (Int): Value inserted at the front of the shape.
last (Int): Value inserted at the back of the shape.

Returns:

IndexList[(rank + Int(2))]