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

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

SRE实战 互联网时代守护先锋,助力企业售后服务体系运筹帷幄!一键直达领取阿里云限量特价优惠。

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

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

 

扫码关注我们
微信号:SRE实战
拒绝背锅 运筹帷幄