tensorflow-tf.concat

匿名 (未验证) 提交于 2019-12-02 23:03:14

tf.concat

tf.concat(



)

按一维连接张量。

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

[D0, D1, ... Raxis, ...Dn]

在此

Raxis = sum(Daxis(i))

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

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

例如:

[[ 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 -*- """ Created on Thu Sep  6 10:16:37 2018  @author: myhaspl """ 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

在Pyhon中, axis的索引是基于0. 正轴在[0, rank(values))中称为轴维。负轴是指axis + rank(values)-th维。
name: 操作名字(可选)
返回:

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

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!