TensorFlow函数:tf.stack
2018-03-19 10:36 更新
tf.stack函数
tf.stack(
values,
axis=0,
name='stack'
)
定义在:tensorflow/python/ops/array_ops.py.
请参阅指南:图层(contrib)>构建神经网络图层的更高级别操作,张量变换>分割和连接
将秩为 R 的张量列表堆叠成一个秩为 (R+1) 的张量.
将 values 中的张量列表打包成一个张量,该张量比 values 中的每个张量都高一个秩,通过沿 axis 维度打包.给定一个形状为(A, B, C)的张量的长度 N 的列表;
如果 axis == 0,那么 output 张量将具有形状(N, A, B, C).如果 axis == 1,那么 output 张量将具有形状(A, N, B, C).
例如:
x = tf.constant([1, 4])
y = tf.constant([2, 5])
z = tf.constant([3, 6])
tf.stack([x, y, z]) # [[1, 4], [2, 5], [3, 6]] (Pack along first dim.)
tf.stack([x, y, z], axis=1) # [[1, 2, 3], [4, 5, 6]]
这与 unpack 相反,numpy 相当于:
tf.stack([x, y, z]) = np.stack([x, y, z])
函数参数:
- values:具有相同形状和类型的 Tensor 对象列表.
- axis:一个 int,要一起堆叠的轴,默认为第一维,负值环绕,所以有效范围是[-(R+1), R+1).
- name:此操作的名称(可选).
函数返回值:
- output:与 values 具有相同的类型的堆叠的 Tensor.
可能引发的异常:
- ValueError:如果 axis 超出范围 [ - (R + 1),R + 1),则引发此异常.