我正在尝试编写自己的卷积神经网络,但至于如何将内核应用于输入我正在努力。
我最初的想法如下(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的所有输出。