tf.FeedForwardNetwork module
tf.FeedForwardNetwork module¶
- class tf.FeedForwardNetwork.FeedForwardNetwork(*args: Any, **kwargs: Any)¶
Bases:
modelzoo.common.layers.tf.BaseLayer.BaseLayerA 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)¶