UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

10 lines (9 loc) 404 B
import { SparseTensor } from '../../../tensor/sparse/tensor'; import { DTypeWasm, WASMTensor } from '../../../tensor/wasm/tensor'; /** * Calculates the sparse-dense matrix product, assuming that a * has zero dense dimensions. * * The result is a dense CPU tensor */ export declare function sparseDenseMatMulWASM<DTpe extends DTypeWasm>(a: SparseTensor<DTpe>, b: WASMTensor<DTpe>): WASMTensor<DTpe>;