如何使用嵌套批处理填充tensorflow占位符

Question

我有一个网络，其输入的特征尺寸为32 假设我有4个输入。批处理大小为128。

所以我有

#include<stdio.h>
int main()
{
   int a[5] = {5, 1, 15, 20, 25};
   int i, j, m;
   i = ++a[1];
   j = a[1]++;
   m = a[i++];
   printf("%d, %d, %d", i, j, m);
}

但是，这给了我一个错误，因为它假定我的网络输入的大小为（4，32）。我希望我的网络输出为input_ph = tf.placeholder(tf.float32, (None, 4, 32), name='ph') network_output = network(input_ph) 。

我可以使用for循环

(batch_size, 4, output_size)

但是这很混乱。在tensorflow中有更好的方法吗？