UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

github.com/Hoff97/tensorjs

Hoff97/tensorjs

4 lines (3 loc) • 281 B
1
2
3
4
import { SparseTensor } from '../../../../tensor/sparse/tensor'; import { DTypeWasm, WASMTensor } from '../../../../tensor/wasm/tensor'; export declare function sumSparseWASM<DTpe extends DTypeWasm>(tensor: SparseTensor<DTpe>, axes: number[], keepDims: boolean): WASMTensor<DTpe>;