Function ANN

rl3/ddpg.py:25–28

(x, layer_sizes, hidden_activation=tf.nn.relu, output_activation=None)

# simple feedforward neural net
def ANN(x, layer_sizes, hidden_activation=tf.nn.relu, output_activation=None):
  for h in layer_sizes[:-1]:
    x = tf.layers.dense(x, units=h, activation=hidden_activation)
  return tf.layers.dense(x, units=layer_sizes[-1], activation=output_activation)


# get all variables within a scope