UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

5 lines (4 loc) 360 B
import { Activation } from '../../library'; import { CPUTensor } from '../../tensor/cpu/tensor'; import { DType } from '../../types'; export declare function conv<DTpe extends DType>(x: CPUTensor<DTpe>, w: CPUTensor<DTpe>, dilations: number[], group: number, pads: number[], strides: number[], activation: Activation, bias?: CPUTensor<DTpe>): CPUTensor<DTpe>;