1 写在前面
tf.concat是很方便的维度拼接API,但在具体使用时,不同维度的拼接有些其他注意事项。
2 API含义(简述)
tf.concat([dataa,datab],dim)
意义分别为:
data:传入需要拼接的两个数据。
dim:指定拼接维度。
3 注意事项
对于深度学习中常用的[b,h,w,c]格式,在指定拼接维度是,索引并非0,1,2,3
tf.concat支持的维度索引为[2,-2]
因此,假如我们需要在[b,h,w,c]的最后一个维度进行拼接,则不能使用3,而是使用-1进行选择。
*——2021.10.24*