def linear_generator(x, hidden_size):
with tf.variable_scope('Generator'):
h0 = tf.nn.softplus(layers.linear(x, hidden_size))
return layers.linear(h0, 1)
微信
公众号
扫码关注公众号