@tensorflow/tfjs-core
Version:
Hardware-accelerated JavaScript library for machine intelligence
6 lines (5 loc) • 595 B
TypeScript
export declare function getReshaped(inputShape: number[], blockShape: number[], prod: number, batchToSpace?: boolean): number[];
export declare function getPermuted(reshapedRank: number, blockShapeRank: number, batchToSpace?: boolean): number[];
export declare function getReshapedPermuted(inputShape: number[], blockShape: number[], prod: number, batchToSpace?: boolean): number[];
export declare function getSliceBeginCoords(crops: number[][], blockShape: number): number[];
export declare function getSliceSize(uncroppedShape: number[], crops: number[][], blockShape: number): number[];