为什么我会使用tf.concat而不是tf.stack？

Question

是否有充分的理由使用tf.concat而不是tf.stack？它们看起来很相似.是否只是为了保证得到的张量与张量输入列表具有相同的维数？

Answer 1

实际上,我误解了它是如何tf.stack运作的.如果axis参数在现有尺寸的范围内,则将在该索引处插入新轴.

例:

import tensorflow as tf

t1 = tf.random_normal([1, 3])
t2 = tf.random_normal([1, 3])

tf.stack([t1, t2], axis=1).shape.as_list() == [1, 2, 3]
tf.concat([t1, t2], axis=1).shape.as_list() == [1, 6]