当堆叠元素很多时，如何在循环中执行张量堆叠？

Question

我正在尝试堆叠张量。我为此目的使用tf.stack（）。但是，我需要堆叠的元素发生了变化。但是，我会知道他们的人数。那么，有没有办法在循环中执行堆栈？

我将我作为形状的张量：[batch_size，256,256,1] 并且我将我作为形状为[256,256,1]的张量。

# I used to do
I = tf.stack([i, i, i, i, i]) # if batch_size is 5
# But I can have a varying value of batch_size.