繁体   English   中英

在Keras中使用2个张量输入定义自定义层

[英]Define a custom layer with 2 tensors inputs in Keras

我想实现一个自定义层。 我的自定义层的2个输入是2个张量,它们来自2个单独的2D卷积层,有一个例子吗?

由于您不需要训练它,因此lambda函数也可以。 或者,您可以保留自定义图层,并将其设置为False。 权重将永远不会针对该层进行更新,并且您在此处所做的任何操作都将向前传播到模型中的下一层,并且如注释中所述,反向传播将通过权重影响其他层。 因此,您的模型肯定会学到一些东西。

我个人建议使用自定义层,如果您稍后决定向该层添加一些知识并检查结果。 您不能在Lambda函数中执行此操作。 如果添加一个(内核),则必须在“调用”方法中使用。 否则,您的模型将在训练过程中引发错误。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM