@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
5 lines (4 loc) • 360 B
TypeScript
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>;