def weight_variable(shape):
    initial = tf.truncated_normal(shape, stddev=0.1)
return tf.Variable(initial)

def bias_variable(shape):
    initial = tf.constant(0.1, shape=shape)
return tf.Variable(initial)
Как исправить ошибку размера в TensorFlow?