UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

4 lines (3 loc) 295 B
import { CPUTensor } from '../../tensor/cpu/tensor'; import { DType } from '../../types'; export declare function pool<DTpe extends DType>(a: CPUTensor<DTpe>, axes: number[], operation: (a: number, b?: number) => number, keepDims: boolean, postProcess?: (a: number) => number): CPUTensor<DTpe>;