将函数应用于tf.split的所有输出的最佳方法是什么?

时间:2018-04-14 17:54:59

标签: tensorflow

我正在尝试编写自己的卷积神经网络,但至于如何将内核应用于输入我正在努力。

我最初的想法如下(psedocode ahead):

a, b, c, d... = tf.split(value, num_or_size_splits=x, 0)
kernel(a)
kernel(b)
kernel(c)
kernel(d)
...

显然必须将它列出来并不适合我。我也不喜欢tf.split没有返回1件作为某种列表的事实,你可以对其进行操作。

最好的方法是什么?我想要一种有效的方法将函数应用于tf.split的所有输出。

0 个答案:

没有答案