自定义Keras Layer

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.

 

posted @ 2020-01-31 18:05  MSTK  阅读(336)  评论(0编辑  收藏  举报