def sigmoid_layer(builder, size):
x = builder.tensor()
m = int(x.get_shape()[1])
n = size
w = tf.Variable(tf.random_uniform([m, n], -1.0, 1.0))
b = tf.Variable(tf.random_uniform([n], -1.0, 1.0))
y = tf.nn.sigmoid(tf.matmul(x, w) + b)
return y.builder()
评论列表
文章目录