def weightVar(shape, mean=0.0, stddev=0.02, name='weights'):
init_w = tf.truncated_normal(shape=shape, mean=mean, stddev=stddev)
return tf.Variable(init_w, name=name)
微信
公众号
扫码关注公众号