CS294-112 DRL HW2记录1

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为输出层



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值