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

Mojo module

pad

Implements tensor padding with constant or edge values for CPU and GPU.

Functions

• ​pad_constant: Fill output with values from input, and edges padded with constant based on paddings.
• ​pad_reflect: Fill output with values from input, and edges padded with reflected values from the unpadded region.
• ​pad_repeat: Fill output with values from input, and edges padded boundary values from the unpadded region.
• ​pad_shape: Compute the output shape of a pad operation, and assert the inputs are compatible.