tf.stack(
values,
axis=0,
name='stack'
)
将秩为 R 的张量列表堆叠成一个秩为 (R+1) 的张量.
将 values 中的张量列表打包成一个张量,该张量比 values 中的每个张量都高一个秩,通过沿 axis 维度打包.给定一个形状为(A, B, C)的张量的长度 N 的列表;
如果 axis == 0,那么 output 张量将具有形状(N, A, B, C).如果 axis == 1,那么 output 张量将具有形状(A, N, B, C).
网友评论