@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
9 lines (8 loc) • 776 B
TypeScript
export declare function getSize(shape: ReadonlyArray<number>, zeroSize?: number): number;
export declare function computeStrides(shape: ReadonlyArray<number>): any[];
export declare function indexToPos(index: ReadonlyArray<number>, strides: ReadonlyArray<number>, shape?: ReadonlyArray<number>): number;
export declare function posToIndex(pos: number, strides: ReadonlyArray<number>): number[];
export declare function compareShapes(a: ReadonlyArray<number>, b: ReadonlyArray<number>): boolean;
export declare function checkEquivShapes(a: ReadonlyArray<number>, b: ReadonlyArray<number>): boolean;
export declare function incrementIndex(index: number[], shape: readonly number[]): void;
export declare function decrementIndex(index: number[], shape: readonly number[]): void;