@hoff97/tensor-js/dist/lib/ops/sparse/matMul/wasm.js

PyTorch like deep learning inferrence library

import { 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 function sparseDenseMatMulWASM(a, b) {
    return new WASMTensor(a.values.wasmTensor.matmul_sparse_dense(a.indices.wasmTensor, b.wasmTensor, a.shape[0]), undefined, a.dtype);
}
//# sourceMappingURL=wasm.js.map