TensorFlow函数:tf.stack

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),则引发此异常.
以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部