擅长:python、mysql、java
<p>通过调用原始的Conv2D构建(您的<code>self.kernel</code>将被替换,然后<code>self.kernelA</code>将永远不会被使用,因此反向传播将永远不会到达它)。在</p>
<p>它也期待偏见和所有常规的东西:</p>
<pre><code>class CustConv2D(Conv2D):
def __init__(self, filters, kernel_size, kernelB=None, activation=None, **kwargs):
#...
#...
#don't use bias if you're not defining it:
super(CustConv2D, self).__init__(self.num_filters, self.kernel_size,
activation=activation,
use_bias=False, **kwargs)
#bonus: don't forget to add the activation to the call above
#it will also replace all your `self.anything` defined before this call
def build(self, input_shape):
#...
#...
#don't use bias:
self.bias = None
#consider the layer built
self.built = True
#do not destroy your build
#comment: super(CustConv2D, self).build(input_shape)
</code></pre>