@ai-on-browser/data-analysis-models/lib/model/nns/layer/maxpool.d.ts

Data analysis model package without any dependencies

/**
 * Max pool layer
 */
export default class MaxPoolLayer extends Layer {
    /**
     * @param {object} config object
     * @param {number | number[]} config.kernel Size of kernel
     * @param {number | number[]} [config.stride] Step of stride
     * @param {number | number[]} [config.padding] size of padding
     * @param {number} [config.channel_dim] Dimension of the channel
     */
    constructor({ kernel, stride, padding, channel_dim, ...rest }: {
        kernel: number | number[];
        stride?: number | number[];
        padding?: number | number[];
        channel_dim?: number;
    });
    _kernel: number | number[];
    _stride: number | number[];
    _padding: number | number[];
    _channel_dim: 1 | -1;
    _index(i: any, c: any, k: any): any[];
    calc(x: any): Tensor<number>;
    _i: any;
    _o: Tensor<number>;
    grad(bo: any): Tensor<number>;
    _bo: any;
    _bi: Tensor<number>;
    toObject(): {
        type: string;
        kernel: number | number[];
        stride: number | number[];
        padding: number | number[];
        channel_dim: number;
    };
}
import Layer from './base.js';
import Tensor from '../../../util/tensor.js';