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 numericalGradient<DTpe extends DType>(input: CPUTensor<DTpe>, fun: (tensor: CPUTensor<DTpe>) => CPUTensor<DTpe>, epsilon?: number): CPUTensor<DTpe>;