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

Data analysis model package without any dependencies

/**
 * Global average pool layer
 */
export default class GlobalAveragePoolLayer extends Layer {
    /**
     * @param {object} config object
     * @param {number} [config.channel_dim] Dimension of the channel
     */
    constructor({ channel_dim, ...rest }: {
        channel_dim?: 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;
        channel_dim: number;
    };
}
import Layer from './base.js';
import Tensor from '../../../util/tensor.js';