Skip to main content
Log in

Mojo function

extend_shape

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

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