Skip to main content
Log in

Mojo module

fold

Implements the fold operation.

Functions

  • fold: Folds array of sliding local blocks into a single output tensor.
  • fold_shape: Returns the shape of the output tensor of the fold operation.