gx

chenyc

2025-06-12 7b72ac13a83764a662159d4a49b7fffb90476ecb

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
/**
 * @license
 * Copyright 2018 Google LLC
 *
 * Use of this source code is governed by an MIT-style
 * license that can be found in the LICENSE file or at
 * https://opensource.org/licenses/MIT.
 * =============================================================================
 */
/// <amd-module name="@tensorflow/tfjs-layers/dist/keras_format/layers/core_serialization" />
import { ActivationSerialization } from '../activation_config';
import { Shape } from '../common';
import { ConstraintSerialization } from '../constraint_config';
import { InitializerSerialization } from '../initializer_config';
import { RegularizerSerialization } from '../regularizer_config';
import { BaseLayerSerialization, LayerConfig } from '../topology_config';
export interface DropoutLayerConfig extends LayerConfig {
    rate: number;
    noise_shape?: number[];
    seed?: number;
}
export type DropoutLayerSerialization = BaseLayerSerialization<'Dropout', DropoutLayerConfig>;
export interface DenseLayerConfig extends LayerConfig {
    units: number;
    activation?: ActivationSerialization;
    use_bias?: boolean;
    input_dim?: number;
    kernel_initializer?: InitializerSerialization;
    bias_initializer?: InitializerSerialization;
    kernel_constraint?: ConstraintSerialization;
    bias_constraint?: ConstraintSerialization;
    kernel_regularizer?: RegularizerSerialization;
    bias_regularizer?: RegularizerSerialization;
    activity_regularizer?: RegularizerSerialization;
}
export type DenseLayerSerialization = BaseLayerSerialization<'Dense', DenseLayerConfig>;
export type FlattenLayerSerialization = BaseLayerSerialization<'Flatten', LayerConfig>;
export interface ActivationLayerConfig extends LayerConfig {
    activation: ActivationSerialization;
}
export type ActivationLayerSerialization = BaseLayerSerialization<'Activation', ActivationLayerConfig>;
export interface RepeatVectorLayerConfig extends LayerConfig {
    n: number;
}
export type RepeatVectorLayerSerialization = BaseLayerSerialization<'RepeatVector', RepeatVectorLayerConfig>;
export interface ReshapeLayerConfig extends LayerConfig {
    target_shape: Shape;
}
export type ReshapeLayerSerialization = BaseLayerSerialization<'Reshape', ReshapeLayerConfig>;
export interface PermuteLayerConfig extends LayerConfig {
    dims: number[];
}
export type PermuteLayerSerialization = BaseLayerSerialization<'Permute', PermuteLayerConfig>;
export interface MaskingLayerConfig extends LayerConfig {
    maskValue: number;
}
export type MaskingLayerSerialization = BaseLayerSerialization<'Masking', MaskingLayerConfig>;
export type CoreLayerSerialization = DropoutLayerSerialization | DenseLayerSerialization | FlattenLayerSerialization | ActivationLayerSerialization | RepeatVectorLayerSerialization | ReshapeLayerSerialization | PermuteLayerSerialization | MaskingLayerSerialization;
export type CoreLayerClassName = CoreLayerSerialization['class_name'];
/**
 * A string array of valid CoreLayer class names.
 *
 * This is guaranteed to match the `CoreLayerClassName` union type.
 */
export declare const coreLayerClassNames: CoreLayerClassName[];