tensorbuilder_patch.py 文件源码

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()