Fork me on GitHub
Arraymancer Technical reference Tutorial Spellbook (How-To's) Under the hood

Module nnp_maxpooling

Search:
Group by:

Procs

proc maxpool2d[T](input: Tensor[T]; kernel: Size2D; padding: Size2D = (0, 0);
                 stride: Size2D = (1, 1)): tuple[max_indices: Tensor[int],
    maxpooled: Tensor[T]] {.
noinit
.}
MaxPool 2D forward pass   Source Edit
proc maxpool2d_backward[T](cached_input_shape: openArray[int] | MetadataArray;
                          cached_max_indices: Tensor[int]; gradOutput: Tensor[T]): Tensor[
    T] {.
noInit
.}
  Source Edit