在tensorflow中使用占位符

Question

input = tf.placeholder(tf.float32, [None, 12, 5])
step = tf.placeholder(tf.int32, [None])

我在tensorflow中使用占位符。输入形状为[None, 12, 5]，步形状为[None]，我想获得外形[None, 5]。步骤中的值介于[0, 11]之间。

Answer 1

您可以使用

step = tf.placeholder(tf.int32, [None, 5])