def create_variable(shape, name, c=None, sigma=None, trainable=True):
if sigma:
initial = tf.truncated_normal(shape, stddev=sigma, name=name)
else:
initial = tf.constant(c if c else 0.0, shape=shape, name=name)
return tf.Variable(initial, trainable=trainable)
评论列表
文章目录