def __init__( self, channel_shared=False, in_channels=None, a_init=truncated_normal(mean=0.0, stddev=0.1), name=None # "prelu" ): super(PRelu, self).__init__(name) self.channel_shared = channel_shared self.in_channels = in_channels self.a_init = a_init if self.channel_shared: self.build((None, )) self._built = True elif self.in_channels is not None: self.build((None, self.in_channels)) self._built = True logging.info("PRelu %s: channel_shared: %s" % (self.name, self.channel_shared))