@tensorflow/tfjs-layers/dist/keras_format/layers/advanced_activation_serialization.d.ts

TensorFlow layers API in JavaScript

/**
 * @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/advanced_activation_serialization" />
import { ConstraintSerialization } from '../constraint_config';
import { InitializerSerialization } from '../initializer_config';
import { RegularizerSerialization } from '../regularizer_config';
import { BaseLayerSerialization, LayerConfig } from '../topology_config';
export interface ReLULayerConfig extends LayerConfig {
    max_value?: number;
}
export type ReLULayerSerialization = BaseLayerSerialization<'ReLU', ReLULayerConfig>;
export interface LeakyReLULayerConfig extends LayerConfig {
    alpha?: number;
}
export type LeakyReLULayerSerialization = BaseLayerSerialization<'LeakyReLU', LeakyReLULayerConfig>;
export interface PReLULayerConfig extends LayerConfig {
    alpha_initializer?: InitializerSerialization;
    alpha_regularizer?: RegularizerSerialization;
    alpha_constraint?: ConstraintSerialization;
    shared_axes?: number | number[];
}
export type PReLULayerSerialization = BaseLayerSerialization<'PReLU', PReLULayerConfig>;
export interface ELULayerConfig extends LayerConfig {
    alpha?: number;
}
export type ELULayerSerialization = BaseLayerSerialization<'ELU', ELULayerConfig>;
export interface ThresholdedReLULayerConfig extends LayerConfig {
    theta?: number;
}
export type ThresholdedReLULayerSerialization = BaseLayerSerialization<'ThresholdedReLU', ThresholdedReLULayerConfig>;
export interface SoftmaxLayerConfig extends LayerConfig {
    axis?: number;
}
export type SoftmaxLayerSerialization = BaseLayerSerialization<'Softmax', SoftmaxLayerConfig>;
export type AdvancedActivationLayerSerialization = ReLULayerSerialization | LeakyReLULayerSerialization | PReLULayerSerialization | ELULayerSerialization | ThresholdedReLULayerSerialization | SoftmaxLayerSerialization;
export type AdvancedActivationLayerClassName = AdvancedActivationLayerSerialization['class_name'];
/**
 * A string array of valid AdvancedActivationLayer class names.
 *
 * This is guaranteed to match the `AdvancedActivationLayerClassName` union
 * type.
 */
export declare const advancedActivationLayerClassNames: AdvancedActivationLayerClassName[];