我试图在纯张量流中制作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