@tensorflow/tfjs-core
Version:
Hardware-accelerated JavaScript library for machine intelligence
9 lines (8 loc) • 414 B
TypeScript
export declare function assertParams(aShape: number[], bShape: number[], axis: number): void;
export declare function computeOutShape(shapes: number[][], axis: number): number[];
export declare function computeGradientSliceShapes(aShape: [number, number], bShape: [number, number]): {
aBegin: [number, number];
aSize: [number, number];
bBegin: [number, number];
bSize: [number, number];
};