使用数组迭代时,Tensorflow压缩结果会有所不同

时间:2018-06-16 07:43:45

标签: arrays tensorflow tensor

enter image description here

我是Tensorflow的新手,我很困惑为什么结果会有所不同。 我认为挤压第二个结果没有应用所以形状是(100,1,20),我期待的是(100,20)。我认为这两个版画做的是同样的事情,但我无法理解它们为什么会产生不同的结果。 谁能给我一个帮助?

这是我用于测试的代码:

with tf.Session() as sess:
    a = tf.zeros((100,6,20))
    print(a.shape)
    print([tf.squeeze(tf.split(a, 6, 1)[0]),tf.squeeze(tf.split(a, 6, 1)[1]),
      tf.squeeze(tf.split(a, 6, 1)[2]),tf.squeeze(tf.split(a, 6, 1)[3])
      ,tf.squeeze(tf.split(a, 6, 1)[4]),tf.squeeze(tf.split(a, 6, 1)[5])])
    print([np.squeeze(t) for t in tf.split(a, 6, 1)] )

0 个答案:

没有答案