@ai-on-browser/data-analysis-models
Version:
Data analysis model package without any dependencies
28 lines (27 loc) • 681 B
TypeScript
/**
* Parametric ReLU layer
*/
export default class ParametricReLULayer extends Layer {
/**
* @param {object} config config
* @param {number | number[] | string} [config.a] a
*/
constructor({ a, ...rest }: {
a?: number | number[] | string;
});
_a: number | Tensor<number>;
_da: number | Tensor<number>;
_aname: string;
calc(x: any): any;
_i: any;
_o: any;
grad(bo: any): any;
_da0: number | Tensor<number>;
update(optimizer: any): void;
toObject(): {
type: string;
a: string | number | Tensor<number>;
};
}
import Layer from './base.js';
import Tensor from '../../../util/tensor.js';