UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

5 lines (4 loc) 418 B
import { CPUTensor } from '../../../tensor/cpu/tensor'; import { SparseTensor } from '../../../tensor/sparse/tensor'; import { DType } from '../../../types'; export declare function aggregateSparseCPU<DTpe extends DType>(tensor: SparseTensor<DTpe>, axes: number[], keepDims: boolean, op: (r: number, e: number) => number, init?: (e: number) => number, postProcess?: (e: number, c: number) => number): CPUTensor<DTpe>;