分类：Flatten层

2020
10-08

Keras实现支持masking的Flatten层代码

不知道为什么，我总是需要实现某种骚操作，而这种骚操作往往是Keras不支持的。例如，我有一个padding过的矩阵，那么它一定是带masking的，然后我想要把它Flatten，再输入到Dense层。然而Keras的Flatten层不支持masking。Keras原本Flatten的实现classFlatten(Layer):def__init__(self,**kwargs):super(Flatten,self).__init__(**kwargs)self.input_spec=InputSpec(min_ndim=3)defcompute_output_shape(self,input_sha... 继续阅读 >