def build_mlp(
input_placeholder,
output_size,
scope,
n_layers=2,
size=64,
activation=tf.tanh,
output_activation=None
):
#========================================================================================#
# ----------SECTION 3----------
# Network building
# Your code should make a feedforward neural network (also called a multilayer perceptron)
# with 'n_layers' hidden layers of size 'size' units.
# The output layer should have size 'output_size' and activation 'output_activation'.
# Hint: use tf.layers.dense
#========================================================================================#
with tf.variable_scope(scope):
# YOUR_CODE_HERE
pass
tf.variable_scope在这里有详细的解释,不是太看得懂,感觉就是用来统一管理、调用下面变量的模块,没有太大实质性的作用。。
这里要创建全连接层,使用tf.layers.dense()来创建,在点击打开链接有比较清晰的解释:
# input layer
out = tf.layers.dense(inputs=input_placeholder, units=size, activation=activation)
#hidden layer
for i in range(n_layers-1):
out=tf.layers.dense(out,units=size,activation=activation)
#output layer
out=tf.layers.dense(out,units=output_size,activation=output_activation)
最后:
return out
out为输出层