2
に複数の値を得た。これは私の畳み込みニューラルネットです:例外TypeError:CONCAT()は、引数 '軸'
frame = tf.placeholder('float', [None, 640-10, 465, 3])
controls_at_each_frame = tf.placeholder('float', [None, 4]) # [w, a, s, d] (1/0)
が使用プレースホルダです
def convolutional_neural_network(frame):
wts = {'conv1': tf.random_normal([5, 5, 3, 32]),
'conv2': tf.random_normal([5, 5, 32, 64]),
'fc': tf.random_normal([158*117*64 + 4, 128]),
'out': tf.random_normal([128, n_classes])
}
biases = {'fc': tf.random_normal([128]),
'out': tf.random_normal([n_classes])
}
conv1 = conv2d(frame, wts['conv1'])
# print(conv1)
conv1 = maxpool2d(conv1)
# print(conv1)
conv2 = conv2d(conv1, wts['conv2'])
conv2 = maxpool2d(conv2)
# print(conv2)
conv2 = tf.reshape(conv2, shape=[-1,158*117*64])
print(conv2)
print(controls_at_each_frame)
conv2 = tf.concat(conv2, controls_at_each_frame, axis=1)
fc = tf.add(tf.matmul(conv2, wts['fc']), biases['fc'])
output = tf.nn.relu(tf.add(tf.matmul(fc, wts['out']), biases['out']))
return output
。
私はGTA San Andreasで自家用車を作っています。私がしたいのは、frame
とcontrols_at_each_frame
を1つのレイヤに連結し、完全に接続されたレイヤに送信することです。私は
conv2 = tf.concat(conv2, controls_at_each_frame, axis=1)
でエラーTypeError: concat() got multiple values for argument 'axis'
を取得し実行すると、あなたは、なぜ、この出来事を説明してもらえますか?
だから私はstackoverflowが大好きです。ちょうど1分で答えを得ました。ありがとうございました。 –
あなたのために働いた場合は受け入れてください! – hausdork
私はすでにしました。 –