ml-basic/dist/types/layers/pooling.d.ts

Lightweight, zero dependency, machine learning library

import Matrix from "../lib/matrix";
import Layer from "./layer";
export type PoolingParams = {
    input: [number, number];
    window: [number, number];
    /**
     * @default window width
     */
    stride?: number;
};
export default abstract class PoolingLayer extends Layer {
    window: [number, number];
    stride: number;
    constructor({ input, window, stride }: PoolingParams);
    abstract backPropagatePoolIndex(aggregate: number, value: number, index: number, indices: number[]): number;
    backPropagate(input: Matrix, _: any, loss: Matrix): Matrix;
}