在CNN中进行张量流初始化后的奇怪形状的内核

时间:2020-04-19 14:05:03

标签: python tensorflow conv-neural-network

我试图在纯张量流中制作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矩阵,但是却给出以下输出:

Output Screen Shot

我完全困惑,因为我想遍历3、2 * 2内核,这是我目前无法做到的。 请帮我了解形状问题!

tensorflow version is 1.14.0

Book's Excerpt

0 个答案:

没有答案