UNPKG

@tensorflow/tfjs-core

Version:

Hardware-accelerated JavaScript library for machine intelligence

github.com/tensorflow/tfjs-core

tensorflow/tfjs-core

4 lines (3 loc) • 283 B
1
2
3
4
export declare function getBroadcastDims(inShape: number[], outShape: number[]): number[]; export declare function getReductionAxes(inShape: number[], outShape: number[]): number[]; export declare function assertAndGetBroadcastShape(shapeA: number[], shapeB: number[]): number[];