Method
__init__
(
self,
channel_shared=False,
in_channels=None,
a_init=truncated_normal(mean=0.0, stddev=0.05),
name=None # "prelu"
)
Source from the content-addressed store, hash-verified
| 54 | """ |
| 55 | |
| 56 | def __init__( |
| 57 | self, |
| 58 | channel_shared=False, |
| 59 | in_channels=None, |
| 60 | a_init=truncated_normal(mean=0.0, stddev=0.05), |
| 61 | name=None # "prelu" |
| 62 | ): |
| 63 | |
| 64 | super(PRelu, self).__init__(name) |
| 65 | self.channel_shared = channel_shared |
| 66 | self.in_channels = in_channels |
| 67 | self.a_init = a_init |
| 68 | |
| 69 | if self.channel_shared: |
| 70 | self.build((None, )) |
| 71 | self._built = True |
| 72 | elif self.in_channels is not None: |
| 73 | self.build((None, self.in_channels)) |
| 74 | self._built = True |
| 75 | |
| 76 | logging.info("PRelu %s: channel_shared: %s" % (self.name, self.channel_shared)) |
| 77 | |
| 78 | def __repr__(self): |
| 79 | s = ('{classname}(') |
Tested by
no test coverage detected