make srelu and pelu work with any initializer

2026-09-10 12:14:52 +08:00 · 2017-04-21 16:47:59 -07:00
parent 278c008440
commit 1b11a5d3dd
1 changed files with 3 additions and 0 deletions
@@ -66,6 +66,7 @@ class PELU(Layer):
                param_shape[i - 1] = 1
                self.param_broadcast[i - 1] = True

+        param_shape = tuple(param_shape)
        # Initialised as ones to emulate the default ELU
        self.alpha = self.add_weight(param_shape,
                                     name='alpha',
@@ -175,6 +176,8 @@ class SReLU(Layer):
                param_shape[i - 1] = 1
                self.param_broadcast[i - 1] = True

+        param_shape = tuple(param_shape)
+
        self.t_left = self.add_weight(param_shape,
                                      name='t_left',
                                      initializer=self.t_left_initializer)