1 |
|
2 |
|
3 |
|
4 |
|
5 |
|
6 |
|
7 |
|
8 |
|
9 |
|
10 |
|
11 | import { fused, serialization, Tensor } from '@tensorflow/tfjs-core';
|
12 | import { Activation } from '../activations';
|
13 | import { Constraint, ConstraintIdentifier } from '../constraints';
|
14 | import { Layer, LayerArgs } from '../engine/topology';
|
15 | import { Initializer, InitializerIdentifier } from '../initializers';
|
16 | import { ActivationIdentifier } from '../keras_format/activation_config';
|
17 | import { DataFormat, InterpolationFormat, PaddingMode, Shape } from '../keras_format/common';
|
18 | import { Regularizer, RegularizerIdentifier } from '../regularizers';
|
19 | import { Kwargs } from '../types';
|
20 | import { LayerVariable } from '../variables';
|
21 |
|
22 |
|
23 |
|
24 |
|
25 |
|
26 | export declare function preprocessConv2DInput(x: Tensor, dataFormat: DataFormat): Tensor;
|
27 |
|
28 |
|
29 |
|
30 |
|
31 |
|
32 | export declare function preprocessConv3DInput(x: Tensor, dataFormat: DataFormat): Tensor;
|
33 |
|
34 |
|
35 |
|
36 |
|
37 |
|
38 |
|
39 |
|
40 |
|
41 |
|
42 |
|
43 |
|
44 |
|
45 |
|
46 |
|
47 |
|
48 |
|
49 | export declare function conv1dWithBias(x: Tensor, kernel: Tensor, bias: Tensor, strides?: number, padding?: string, dataFormat?: DataFormat, dilationRate?: number): Tensor;
|
50 |
|
51 |
|
52 |
|
53 |
|
54 |
|
55 |
|
56 |
|
57 |
|
58 |
|
59 |
|
60 |
|
61 |
|
62 | export declare function conv1d(x: Tensor, kernel: Tensor, strides?: number, padding?: string, dataFormat?: DataFormat, dilationRate?: number): Tensor;
|
63 |
|
64 |
|
65 |
|
66 |
|
67 |
|
68 |
|
69 |
|
70 |
|
71 |
|
72 |
|
73 | export declare function conv2d(x: Tensor, kernel: Tensor, strides?: number[], padding?: string, dataFormat?: DataFormat, dilationRate?: [number, number]): Tensor;
|
74 |
|
75 |
|
76 |
|
77 |
|
78 |
|
79 | export declare function conv2dWithBiasActivation(x: Tensor, kernel: Tensor, bias: Tensor, strides?: number[], padding?: string, dataFormat?: DataFormat, dilationRate?: [number, number], activation?: fused.Activation): Tensor;
|
80 |
|
81 |
|
82 |
|
83 |
|
84 |
|
85 |
|
86 |
|
87 |
|
88 |
|
89 |
|
90 | export declare function conv3d(x: Tensor, kernel: Tensor, strides?: number[], padding?: string, dataFormat?: DataFormat, dilationRate?: [number, number, number]): Tensor;
|
91 |
|
92 |
|
93 |
|
94 |
|
95 |
|
96 | export declare function conv3dWithBias(x: Tensor, kernel: Tensor, bias: Tensor, strides?: number[], padding?: string, dataFormat?: DataFormat, dilationRate?: [number, number, number]): Tensor;
|
97 |
|
98 |
|
99 |
|
100 | export declare interface BaseConvLayerArgs extends LayerArgs {
|
101 | |
102 |
|
103 |
|
104 |
|
105 | kernelSize: number | number[];
|
106 | |
107 |
|
108 |
|
109 |
|
110 |
|
111 |
|
112 |
|
113 | strides?: number | number[];
|
114 | |
115 |
|
116 |
|
117 | padding?: PaddingMode;
|
118 | |
119 |
|
120 |
|
121 |
|
122 |
|
123 |
|
124 |
|
125 |
|
126 |
|
127 |
|
128 |
|
129 |
|
130 | dataFormat?: DataFormat;
|
131 | |
132 |
|
133 |
|
134 |
|
135 |
|
136 |
|
137 |
|
138 | dilationRate?: number | [number] | [number, number] | [number, number, number];
|
139 | |
140 |
|
141 |
|
142 |
|
143 |
|
144 | activation?: ActivationIdentifier;
|
145 | |
146 |
|
147 |
|
148 | useBias?: boolean;
|
149 | |
150 |
|
151 |
|
152 | kernelInitializer?: InitializerIdentifier | Initializer;
|
153 | |
154 |
|
155 |
|
156 | biasInitializer?: InitializerIdentifier | Initializer;
|
157 | |
158 |
|
159 |
|
160 | kernelConstraint?: ConstraintIdentifier | Constraint;
|
161 | |
162 |
|
163 |
|
164 | biasConstraint?: ConstraintIdentifier | Constraint;
|
165 | |
166 |
|
167 |
|
168 | kernelRegularizer?: RegularizerIdentifier | Regularizer;
|
169 | |
170 |
|
171 |
|
172 | biasRegularizer?: RegularizerIdentifier | Regularizer;
|
173 | |
174 |
|
175 |
|
176 | activityRegularizer?: RegularizerIdentifier | Regularizer;
|
177 | }
|
178 |
|
179 |
|
180 |
|
181 |
|
182 |
|
183 | export declare interface ConvLayerArgs extends BaseConvLayerArgs {
|
184 | |
185 |
|
186 |
|
187 |
|
188 | filters: number;
|
189 | }
|
190 |
|
191 |
|
192 |
|
193 | export declare abstract class BaseConv extends Layer {
|
194 | protected readonly rank: number;
|
195 | protected readonly kernelSize: number[];
|
196 | protected readonly strides: number[];
|
197 | protected readonly padding: PaddingMode;
|
198 | protected readonly dataFormat: DataFormat;
|
199 | protected readonly activation: Activation;
|
200 | protected readonly useBias: boolean;
|
201 | protected readonly dilationRate: number[];
|
202 | protected readonly biasInitializer?: Initializer;
|
203 | protected readonly biasConstraint?: Constraint;
|
204 | protected readonly biasRegularizer?: Regularizer;
|
205 | protected bias: LayerVariable;
|
206 | readonly DEFAULT_KERNEL_INITIALIZER: InitializerIdentifier;
|
207 | readonly DEFAULT_BIAS_INITIALIZER: InitializerIdentifier;
|
208 | constructor(rank: number, args: BaseConvLayerArgs);
|
209 | protected static verifyArgs(args: BaseConvLayerArgs): void;
|
210 | getConfig(): serialization.ConfigDict;
|
211 | }
|
212 | /**
|
213 | * Abstract nD convolution layer. Ancestor of convolution layers which reduce
|
214 | * across channels, i.e., Conv1D and Conv2D, but not DepthwiseConv2D.
|
215 | */
|
216 | export declare abstract class Conv extends BaseConv {
|
217 | protected readonly filters: number;
|
218 | protected kernel: LayerVariable;
|
219 | protected readonly kernelInitializer?: Initializer;
|
220 | protected readonly kernelConstraint?: Constraint;
|
221 | protected readonly kernelRegularizer?: Regularizer;
|
222 | constructor(rank: number, args: ConvLayerArgs);
|
223 | build(inputShape: Shape | Shape[]): void;
|
224 | call(inputs: Tensor | Tensor[], kwargs: Kwargs): Tensor | Tensor[];
|
225 | computeOutputShape(inputShape: Shape | Shape[]): Shape | Shape[];
|
226 | getConfig(): serialization.ConfigDict;
|
227 | protected static verifyArgs(args: ConvLayerArgs): void;
|
228 | }
|
229 | export declare class Conv2D extends Conv {
|
230 |
|
231 | static className: string;
|
232 | constructor(args: ConvLayerArgs);
|
233 | getConfig(): serialization.ConfigDict;
|
234 | protected static verifyArgs(args: ConvLayerArgs): void;
|
235 | }
|
236 | export declare class Conv3D extends Conv {
|
237 |
|
238 | static className: string;
|
239 | constructor(args: ConvLayerArgs);
|
240 | getConfig(): serialization.ConfigDict;
|
241 | protected static verifyArgs(args: ConvLayerArgs): void;
|
242 | }
|
243 | export declare class Conv2DTranspose extends Conv2D {
|
244 |
|
245 | static className: string;
|
246 | constructor(args: ConvLayerArgs);
|
247 | build(inputShape: Shape | Shape[]): void;
|
248 | call(inputs: Tensor | Tensor[], kwargs: Kwargs): Tensor | Tensor[];
|
249 | computeOutputShape(inputShape: Shape | Shape[]): Shape | Shape[];
|
250 | getConfig(): serialization.ConfigDict;
|
251 | }
|
252 | export declare class Conv3DTranspose extends Conv3D {
|
253 |
|
254 | static className: string;
|
255 | constructor(args: ConvLayerArgs);
|
256 | build(inputShape: Shape | Shape[]): void;
|
257 | call(inputs: Tensor | Tensor[], kwargs: Kwargs): Tensor | Tensor[];
|
258 | computeOutputShape(inputShape: Shape | Shape[]): Shape | Shape[];
|
259 | getConfig(): serialization.ConfigDict;
|
260 | }
|
261 | export declare interface SeparableConvLayerArgs extends ConvLayerArgs {
|
262 | |
263 |
|
264 |
|
265 |
|
266 |
|
267 |
|
268 | depthMultiplier?: number;
|
269 | |
270 |
|
271 |
|
272 | depthwiseInitializer?: InitializerIdentifier | Initializer;
|
273 | |
274 |
|
275 |
|
276 | pointwiseInitializer?: InitializerIdentifier | Initializer;
|
277 | |
278 |
|
279 |
|
280 | depthwiseRegularizer?: RegularizerIdentifier | Regularizer;
|
281 | |
282 |
|
283 |
|
284 | pointwiseRegularizer?: RegularizerIdentifier | Regularizer;
|
285 | |
286 |
|
287 |
|
288 | depthwiseConstraint?: ConstraintIdentifier | Constraint;
|
289 | |
290 |
|
291 |
|
292 | pointwiseConstraint?: ConstraintIdentifier | Constraint;
|
293 | }
|
294 | export declare class SeparableConv extends Conv {
|
295 |
|
296 | static className: string;
|
297 | readonly depthMultiplier: number;
|
298 | protected readonly depthwiseInitializer?: Initializer;
|
299 | protected readonly depthwiseRegularizer?: Regularizer;
|
300 | protected readonly depthwiseConstraint?: Constraint;
|
301 | protected readonly pointwiseInitializer?: Initializer;
|
302 | protected readonly pointwiseRegularizer?: Regularizer;
|
303 | protected readonly pointwiseConstraint?: Constraint;
|
304 | readonly DEFAULT_DEPTHWISE_INITIALIZER: InitializerIdentifier;
|
305 | readonly DEFAULT_POINTWISE_INITIALIZER: InitializerIdentifier;
|
306 | protected depthwiseKernel: LayerVariable;
|
307 | protected pointwiseKernel: LayerVariable;
|
308 | constructor(rank: number, config?: SeparableConvLayerArgs);
|
309 | build(inputShape: Shape | Shape[]): void;
|
310 | call(inputs: Tensor | Tensor[], kwargs: Kwargs): Tensor | Tensor[];
|
311 | getConfig(): serialization.ConfigDict;
|
312 | }
|
313 | export declare class SeparableConv2D extends SeparableConv {
|
314 |
|
315 | static className: string;
|
316 | constructor(args?: SeparableConvLayerArgs);
|
317 | }
|
318 | export declare class Conv1D extends Conv {
|
319 |
|
320 | static className: string;
|
321 | constructor(args: ConvLayerArgs);
|
322 | getConfig(): serialization.ConfigDict;
|
323 | protected static verifyArgs(args: ConvLayerArgs): void;
|
324 | }
|
325 | export declare interface Cropping2DLayerArgs extends LayerArgs {
|
326 | |
327 |
|
328 |
|
329 |
|
330 |
|
331 |
|
332 |
|
333 |
|
334 |
|
335 |
|
336 |
|
337 |
|
338 | cropping: number | [number, number] | [[number, number], [number, number]];
|
339 | |
340 |
|
341 |
|
342 |
|
343 |
|
344 |
|
345 |
|
346 |
|
347 |
|
348 |
|
349 |
|
350 |
|
351 | dataFormat?: DataFormat;
|
352 | }
|
353 | export declare class Cropping2D extends Layer {
|
354 |
|
355 | static className: string;
|
356 | protected readonly cropping: [[number, number], [number, number]];
|
357 | protected readonly dataFormat: DataFormat;
|
358 | constructor(args: Cropping2DLayerArgs);
|
359 | computeOutputShape(inputShape: Shape): Shape;
|
360 | call(inputs: Tensor | Tensor[], kwargs: Kwargs): Tensor | Tensor[];
|
361 | getConfig(): serialization.ConfigDict;
|
362 | }
|
363 | export declare interface UpSampling2DLayerArgs extends LayerArgs {
|
364 | |
365 |
|
366 |
|
367 |
|
368 |
|
369 | size?: number[];
|
370 | |
371 |
|
372 |
|
373 |
|
374 |
|
375 |
|
376 |
|
377 |
|
378 |
|
379 |
|
380 |
|
381 |
|
382 | dataFormat?: DataFormat;
|
383 | |
384 |
|
385 |
|
386 |
|
387 | interpolation?: InterpolationFormat;
|
388 | }
|
389 | export declare class UpSampling2D extends Layer {
|
390 |
|
391 | static className: string;
|
392 | protected readonly DEFAULT_SIZE: number[];
|
393 | protected readonly size: number[];
|
394 | protected readonly dataFormat: DataFormat;
|
395 | protected readonly interpolation: InterpolationFormat;
|
396 | constructor(args: UpSampling2DLayerArgs);
|
397 | computeOutputShape(inputShape: Shape): Shape;
|
398 | call(inputs: Tensor | Tensor[], kwargs: Kwargs): Tensor | Tensor[];
|
399 | getConfig(): serialization.ConfigDict;
|
400 | }
|