@tensorflow/tfjs-core
Version:
Hardware-accelerated JavaScript library for machine intelligence
5 lines (4 loc) • 352 B
TypeScript
export declare function getBroadcastDims(inShape: number[], outShape: number[]): number[];
export declare function getReductionAxes(inShape: number[], outShape: number[]): number[];
export declare function broadcastDimsAreOuter(dims: number[]): boolean;
export declare function assertAndGetBroadcastShape(shapeA: number[], shapeB: number[]): number[];