UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

4 lines (3 loc) 263 B
import { CPUTensor } from '../../tensor/cpu/tensor'; import { DType } from '../../types'; export declare function averagePool<DTpe extends DType>(x: CPUTensor<DTpe>, kernelShape: number[], pads: number[], strides: number[], includePad: boolean): CPUTensor<DTpe>;