Skip to main content
Log in

Mojo function

extend_shape

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

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