【发布时间】:2021-04-22 00:11:26
【问题描述】:
这个问题很简单。我想将我的神经网络中的单层乘以一个数字。
这个数字是n = max(abs(input)),所以实际输入是input/n,而输出是output*n。我想知道是否有办法将这些计算插入到我的神经网络模型中。
在旁注中,我想这样做是因为我在某处读到一个常见的解决方案是对输入进行规范化,然后对输出进行去规范化,这就是我想要做的。然而,在我的多输入神经网络中,一个输入是二进制数据,另一个是常规实数,所以我只想对这个实数输入进行归一化,并对特定的下游层进行反归一化。
【问题讨论】:
-
您可以通过继承 Keras 层类来创建自定义层,如此处所述 - Keras Custom Layer 并在您的神经网络中使用它,或者您可以使用 Lambda 层,如此处所述 - @987654322 @.
标签: python tensorflow neural-network