def weight_variable(shape,name):
initial = tf.truncated_normal(shape, stddev=1.0 / math.sqrt(float(shape[0])))
return tf.Variable(initial, name=name)
微信
公众号
扫码关注公众号