Tensorflow如何实现多维矩阵的与拼接操作示例？

本文共计358个文字，预计阅读时间需要2分钟。

最近在使用tensorflow进行网络训练时，需要提取出表现良好的卷积核的部分层数据。由于tensorflow中的tensor和python中的list不同，无法直接使用加法进行拼接。后来发现了一个函数可以实现这一功能。

最近在使用tensorflow进行网络训练的时候，需要提取出别人训练好的卷积核的部分层的数据。由于tensorflow中的tensor和python中的list不同，无法直接使用加法进行拼接，后来发现一个函数可以完成tensor的拼接。

函数形式如下：

tf.concat(concat_dim,values,name='concat')

其中，第一个参数表示需要拼接的多维tensor，并且可以将多个tensor同事拼接，第二个表示按照哪一个维度拼接（从数字0开始）。

例子：创建一个三维的tensor，然后分别取出最后一个维度（注意：tensor支持与python中list相似的切片操作，可以使用这种方式进行拆分），然后在拼接在一起。