DriftTexture · function

maxPool2d

out[channels][oh][ow], each the largest of its kernel-square window at stride, with oh = ⌊(h − kernel)/stride⌋ + 1: no padding, and a ragged last row or column dropped, as PyTorch's max_pool2d does without ceil_mode.

Explained in DriftTexture.

function maxPool2d(out: Float32Array, input: Float32Array, channels: number, h: number, w: number, kernel: number, stride: number): void
import { maxPool2d } from '@driftengine/texture';

Parameters

ParameterTypeDescription
outFloat32Array
inputFloat32Array
channelsnumber
hnumber
wnumber
kernelnumber
stridenumber