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)>