@ai-on-browser/data-analysis-models
Version:
Data analysis model package without any dependencies
24 lines (23 loc) • 453 B
TypeScript
/**
* Log softmax layer
*/
export default class LogSoftmaxLayer extends Layer {
/**
* @param {object} config object
* @param {number} [config.axis] Axis
*/
constructor({ axis, ...rest }: {
axis?: number;
});
_axis: number;
calc(x: any): any;
_s: any;
_o: any;
grad(bo: any): any;
_bi: any;
toObject(): {
type: string;
axis: number;
};
}
import Layer from './base.js';