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 function

max_pool_shape

def max_pool_shape(input: T, filter: T, strides: T, dilations: T, paddings: T) -> IndexList[T.rank]

Computes the output shape for the mo.max_pool graph op.

Args:

• ​input (T): Rank-4 batched image input to the pooling operator.
• ​filter (T): One-dimensional tensor of filter sizes on the height and width dimensions, (filter_h, filter_w).
• ​strides (T): One-dimensional tensor of strides on the height and width dimensions, (stride_h, stride_w).
• ​dilations (T): One-dimensional tensor of dilations on the height and width dimensions, (dilation_h, dilation_w).
• ​paddings (T): One-dimensional tensor of paddings on the height and width dimensions, (pad_h_before, pad_h_after, pad_w_before, pad_w_after).

Returns:

IndexList[T.rank]: The output shape of the max pooling operation.