@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
6 lines (5 loc) • 548 B
TypeScript
import { CPUTensor } from '../../../tensor/cpu/tensor';
import { SparseTensor } from '../../../tensor/sparse/tensor';
import { DType } from '../../../types';
export declare function binaryDenseCPU<DTpe extends DType>(a: SparseTensor<DTpe>, b: CPUTensor<DTpe>, resultShape: readonly number[], op: (a: number, b: number) => number): SparseTensor<DTpe>;
export declare function binarySparseCPU<DTpe extends DType>(a: SparseTensor<DTpe>, b: SparseTensor<DTpe>, resultShape: readonly number[], op: (a: number, b: number) => number): SparseTensor<DTpe>;