import { Module } from "./nn_module";
export declare class AvgPooling2d extends Module {
}
export declare class Conv2d extends Module {
    inChannels: number;
    outChannels: number;
    constructor(inChannels: number, outChannels: number, kernelSize: number | [number, number], stride: number | [number, number], padding: number | [number, number] | "valid" | "same", dtype: string);
}
export declare class ConvTranspose2d extends Module {
}
export declare class GroupNorm extends Module {
    numGroups: number;
    numChannels: number;
    constructor(numGroups: number, numChannels: number);
}
export declare class Linear extends Module {
    inChannels: number;
    outChannels: number;
    constructor(inChannels: number, outChannels: number);
}