UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

5 lines (4 loc) 496 B
import { SparseTensor } from '../../../../tensor/sparse/tensor'; import { DTypeWasm, WASMTensor } from '../../../../tensor/wasm/tensor'; export declare function multiplyDenseWASM<DTpe extends DTypeWasm>(a: SparseTensor<DTpe>, b: WASMTensor<DTpe>, resultShape: readonly number[], alpha: number): SparseTensor<DTpe>; export declare function multiplySparseWASM<DTpe extends DTypeWasm>(a: SparseTensor<DTpe>, b: SparseTensor<DTpe>, resultShape: readonly number[], alpha: number): SparseTensor<DTpe>;