UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

11 lines (10 loc) 424 B
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>;