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))
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))
