拆栈时保持TensorArray大小

时间:2018-10-10 22:12:38

标签: python tensorflow

我正在使用tensorflow 1.11.0。

当我使用TensorArray的unstack方法时,我不希望TensorArray的大小发生变化。可以这样做吗?

我在下面发布了一个示例。

import tensorflow as tf
tf.enable_eager_execution()

# I want the size to be 4.
ta = tf.TensorArray(dtype=tf.float32, size=4)
assert(ta.size().numpy() == 4)

# The size will become 2 after I call unstack.
xs = tf.constant([1, 2], dtype=tf.float32)
ta = ta.unstack(xs)
assert(ta.size().numpy() == 2)

0 个答案:

没有答案