lcd.py 文件源码

def buildFeatures(self, type='shared'):
        assert self.checkTensor('q+')
        assert self.checkTensor('q-')
        assert self.checkTensor('a+')
        assert self.checkTensor('a-')
        srelu = lambda name: SReLU(name=name)
        features = []
        if type == 'shared':
            q_features = Merge(
                mode='concat',
                name='q-features',
            )([
                self.linkFeature('q+', 'shared-convolution', activation=srelu),
                self.linkFeature('q-', 'shared-convolution', activation=srelu)
            ])
            a_features = Merge(
                mode='concat',
                name='a-features',
            )([
                self.linkFeature('a+', 'shared-convolution', activation=srelu),
                self.linkFeature('a-', 'shared-convolution', activation=srelu)
            ])
        else:
            raise Error('Not Supported')
        self.tensors['q-features'] = q_features
        self.tensors['a-features'] = a_features