tf.FeedForwardNetwork module

tf.FeedForwardNetwork module

class tf.FeedForwardNetwork.FeedForwardNetwork(*args: Any, **kwargs: Any)

Bases: modelzoo.common.layers.tf.BaseLayer.BaseLayer

A feed forward network that consists of a stack of fully connected layers.

Parameters
  • layers_units (int) – List of units for each layer.

  • layers_activation (str) – List of activation types (str) for each layer.

  • layers_dropout_rates (float) – List of dropout rates (float) for each layer.

  • use_bias (bool) – If True, use bias throughout all layers.

  • kernel_initializer – Kernel initializer. Defaults to "glorot_uniform".

  • bias_initializer (callable) – Bias initializer. Defaults to "zeros".

  • kernel_regularizer (callable) – Kernel regularizer.

  • bias_initializer – Bias regularizer.

call(inputs, training=True, **kwargs)