tf_agents.keras_layers.BiasLayer

Keras layer that only adds a bias to the input.

BiasLayer implements the operation: output = input + bias

bias_initializer Initializer for the bias vector.

Input shape:

nD tensor with shape: (batch_size, ..., input_dim). The most common situation would be a 2D input with shape (batch_size, input_dim). Note a rank of at least 2 is required.

Output shape:

nD tensor with shape: (batch_size, ..., input_dim). For instance, for a 2D input with shape (batch_size, input_dim), the output would have shape (batch_size, input_dim).