def weight_variable(self, shape, name):
initial = tf.truncated_normal(shape, stddev=0.01)
return tf.get_variable(name=name, initializer=initial, trainable=True)
微信
公众号
扫码关注公众号