@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
11 lines (10 loc) • 424 B
TypeScript
import { CPUTensor } from '../../../tensor/cpu/tensor';
import { SparseTensor } from '../../../tensor/sparse/tensor';
import { DType } from '../../../types';
/**
* Calculates the sparse-dense matrix product, assuming that a
* has zero dense dimensions.
*
* The result is a dense CPU tensor
*/
export declare function sparseDenseMatMulCPU<DTpe extends DType>(a: SparseTensor<DTpe>, b: CPUTensor<DTpe>): CPUTensor<DTpe>;