首页 > 其他 > 详细

自定义Keras Layer

时间:2020-01-31 20:24:15      阅读:46      评论:0      收藏:0      [点我收藏+]

Keras的Layer其实就是一个Class, 要具有以下几个方法:

(1) build(input_shape): 定义权重的地方, 如果不需要定义权重, 也要有self.built = True;

(2) call(x): 编写功能逻辑的地方, 实现Layer的功能逻辑;

(3) compute_output_shape(input_shape): 根据input_shape, 计算形状变化的地方, 能够让Keras推断输出的Tensor的Shape.

 

自定义Keras Layer

原文:https://www.cnblogs.com/mstk/p/12246126.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!