Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

Module: tf_agents.networks.lstm_encoding_network

View source on GitHub

Keras LSTM Encoding Network.

Implements a network that will generate the following layers:

[optional]: preprocessing_layers # preprocessing_layers [optional]: (Add | Concat(axis=-1) | ...) # preprocessing_combiner [optional]: Conv2D # input_conv_layer_params Flatten [optional]: Dense # input_fc_layer_params [optional]: LSTM cell [optional]: Dense # output_fc_layer_params

Classes

class LSTMEncodingNetwork: Recurrent network.

Other Members

  • KERAS_LSTM_FUSED_IMPLEMENTATION = 2