Я пытался сделать CNN в чистом тензорном потоке, и для инициализации ядер для слоя я использую следующий код:
import tensorflow as tf
kernel = tf.Variable(tf.random.truncated_normal([2,2,1,3], stddev=0.1))
init = tf.initialize_all_variables()
with tf.Session() as sess:
sess.run(init)
a_value = sess.run(kernel)
print(a_value.shape)
print(a_value)
Я ожидал, что он даст 3, 2 * 2 матрицы но вместо этого он дал следующий вывод:
Снимок экрана вывода
Я совершенно сбит с толку, поскольку хотел перебрать 3, 2 * 2 ядра, которыми я являюсь не в состоянии сделать в настоящее время. Пожалуйста, помогите мне понять проблему формы!
tensorflow version is 1.14.0
Отрывок книги