UNPKG

@ai-on-browser/data-analysis-models

Version:

Data analysis model package without any dependencies

32 lines (31 loc) 779 B
/** * Global Lp pool layer */ export default class GlobalLpPoolLayer extends Layer { /** * @param {object} config object * @param {number} [config.p] p * @param {number} [config.channel_dim] Dimension of the channel */ constructor({ p, channel_dim, ...rest }: { p?: number; channel_dim?: number; }); _p: number; _channel_dim: 1 | -1; _index(i: any, c: any, k: any): any[]; calc(x: any): Tensor<number>; _i: any; _sp: Tensor<number>; _o: Tensor<number>; grad(bo: any): Tensor<number>; _bo: any; _bi: Tensor<number>; toObject(): { type: string; p: number; channel_dim: number; }; } import Layer from './base.js'; import Tensor from '../../../util/tensor.js';