@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
11 lines • 837 B
JavaScript
import { SparseTensor } from '../../../../tensor/sparse/tensor';
import { WASMTensor } from '../../../../tensor/wasm/tensor';
export function subtractDenseWASM(a, b, resultShape, alpha, beta) {
const vals = new WASMTensor(a.values.wasmTensor.subtract_sparse_dense(a.indices.wasmTensor, b.wasmTensor, new Uint32Array(resultShape), alpha, beta), undefined, a.dtype);
return new SparseTensor(vals, a.indices.copy(), resultShape, a.denseDims);
}
export function subtractSparseWASM(a, b, resultShape, alpha, beta) {
const vals = new WASMTensor(a.values.wasmTensor.subtract_sparse_sparse(a.indices.wasmTensor, b.indices.wasmTensor, b.values.wasmTensor, new Uint32Array(resultShape), alpha, beta), undefined, a.dtype);
return new SparseTensor(vals, a.indices.copy(), resultShape, a.denseDims);
}
//# sourceMappingURL=wasm.js.map