golbin / TensorFlow-Tutorials

텐서플로우를 기초부터 응용까지 단계별로 연습할 수 있는 소스 코드를 제공합니다

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

안녕하세요 reshape 함수에 대해 도움을 주시면 감사하겠습니다.

smj21 opened this issue · comments

commented

올려주신 코드를 보고 연습을 하고 있는데 다음과 같은 문제가 생깁니다.
코드:
for i in range(total_batch):
batch_xs, batch_ys = data.next_batch(BatchSize=batch_size, shuffle=data_shuffle)
batch_xs = np.asarray(batch_xs)
batch_ys = np.asarray(batch_ys)
batch_xs = batch_xs.reshape(-1, 48, 48, 1)

    _, cost_val = sess.run([optimizer, cost],
                           feed_dict={X: batch_xs,Y: batch_ys,keep_prob: 0.7})
    total_cost += cost_val
    data_shuffle = False

*batch size = 50입니다.
X = tf.placeholder(tf.float32, [None, 48, 48, 1])
Y = tf.placeholder(tf.float32, [None, 7])
keep_prob = tf.placeholder(tf.float32)

이러식으로 돌렸을 때, 오류문이 다음과 같이 나옵니다.
ValueError : cannot reshape array of size 50 into shape (48,48,1)

현재 파이썬3에서 위 코드를 돌리고 있는데, 버전이 달라서 reshape 함수가 안돌아가는 건가요?