tensorflow随笔 -tf.concat

最新推荐文章于 2020-11-25 20:50:42 发布

「已注销」

最新推荐文章于 2020-11-25 20:50:42 发布

阅读量224

点赞数

CC 4.0 BY-SA版权

文章标签： tensorflow

本文链接：https://blog.youkuaiyun.com/AI_LX/article/details/89212849

本文详细介绍了TensorFlow中的tf.concat函数，该函数用于在一维上连接多个张量。文章通过实例展示了如何使用tf.concat来处理不同维度的张量，并解释了正负轴在Python中的使用方式。此外，还对比了tf.concat与tf.stack在操作上的区别。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

tf.concat

tf.concat(
values,
axis,
name=‘concat’
)

按一维连接张量。

沿着维度轴连接张量值的列表。如果values[i].shape=[D0, D1, … Daxis(i), …Dn]，则连接的结果具有形状如下：

[D0, D1, … Raxis, …Dn]

在此

Raxis = sum(Daxis(i))

也就是说，来自输入张量的数据沿着轴维度连接。

输入张量的维数必须匹配，除轴外的所有维度必须相等。

例如：

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

t1 = [[1, 2, 3], [4, 5, 6]]
t2 = [[7, 8, 9], [10, 11, 12]]
a=tf.concat([t1, t2], 0)  
b=tf.concat([t1, t2], 1)  
sess=tf.Session()
with sess:
    print sess.run(a)
    print sess.run(b)

[[ 1 2 3]
[ 4 5 6]
[ 7 8 9]
[10 11 12]]
[[ 1 2 3 7 8 9]
[ 4 5 6 10 11 12]]

在python中，axis可以为负值。负轴（axis）x被解释为从秩(rank)的末尾开始计数，即axis+rank(values)-Th维。

例如：

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import tensorflow as tf
t1 = [[[1, 2], [2, 3]], [[4, 4], [5, 3]]]
t2 = [[[7, 4], [8, 4]], [[2, 10], [15, 11]]]
a=tf.concat([t1, t2], -1)
sess=tf.Session()
with sess:
    print sess.run(a)

[[[ 1 2 7 4]
[ 2 3 8 4]]

[[ 4 4 2 10]
[ 5 3 15 11]]]

注意：如果你在一个新的轴上连接，考虑使用堆栈。例如。

tf.concat([tf.expand_dims(t, axis) for t in tensors], axis)

可写成

tf.stack(tensors, axis=axis)

参数:

values: 一个Tensor对象的列表或单个tensor
axis: 0-D int32 Tensor.沿着该轴连接，必须在 [-rank(values), rank(values))
在Pyhon中， axis的索引是基于0. 正轴在[0, rank(values))中称为轴维。负轴是指axis + rank(values)-th维。
name: 操作名字（可选）
返回:

从input tensors连接而成的Tensor结果。