Python tf.TensorArray.stack用法及代码示例

用法

stack(
    name=None
)

将 TensorArray 中的值作为堆叠的 Tensor 返回。

所有值都必须已写入，并且它们的形状必须全部匹配。如果输入形状具有排名- R ，则输出形状将具有排名- (R+1) 。

例如：

ta = tf.TensorArray(tf.int32, size=3)
ta.write(0, tf.constant([1, 2]))
ta.write(1, tf.constant([3, 4]))
ta.write(2, tf.constant([5, 6]))
ta.stack()
<tf.Tensor:shape=(3, 2), dtype=int32, numpy=
array([[1, 2],
       [3, 4],
       [5, 6]], dtype=int32)>