Skip to main content

/

Docs

Nightly

v24.6

Log in

Mojo module

convolution

Ops commonly used in convolutional networks.

Functions​

  • ​avg_pool: Computes average pooling with the given filter shape, strides, and dilations.
  • ​conv2d: Computes the 2-D convolution product of the input with the given filter, strides, dilations, paddings, and groups.
  • ​conv3d: Computes the 3-D convolution product of the input with the given filter, strides, dilations, paddings, and groups.
  • ​max_pool: Computes max pooling with the given filter shape, strides, and dilations.