@ai-on-browser/data-analysis-models
Version:
Data analysis model package without any dependencies
23 lines (22 loc) • 496 B
TypeScript
/**
* Reshape layer
*/
export default class ReshapeLayer extends Layer {
/**
* @param {object} config config
* @param {number[] | string} config.size Size of output
*/
constructor({ size, ...rest }: {
size: number[] | string;
});
_size: string | number[];
calc(x: any): any;
_in_size: any;
_out_size: any;
grad(bo: any): any;
toObject(): {
type: string;
size: string | number[];
};
}
import Layer from './base.js';