fix Bias layer

2017-08-02 11:37:39 +00:00 · 2017-08-02 11:37:39 +00:00 · 7ac67fba8f
parent 049d966710
commit 7ac67fba8f
1 changed files with 1 additions and 1 deletions
--- a/onn_core.py
+++ b/onn_core.py
@ -840,7 +840,7 @@ class Bias(Layer):
        shape = parent.output_shape
        self.input_shape = shape
        self.output_shape = shape
-        self.biases.shape = (self.dim,)
+        self.biases.shape = (shape[-1],)

    def forward(self, X):
        return X + self.biases.f