fix Bias layer
This commit is contained in:
parent
049d966710
commit
7ac67fba8f
1 changed files with 1 additions and 1 deletions
|
@ -840,7 +840,7 @@ class Bias(Layer):
|
||||||
shape = parent.output_shape
|
shape = parent.output_shape
|
||||||
self.input_shape = shape
|
self.input_shape = shape
|
||||||
self.output_shape = shape
|
self.output_shape = shape
|
||||||
self.biases.shape = (self.dim,)
|
self.biases.shape = (shape[-1],)
|
||||||
|
|
||||||
def forward(self, X):
|
def forward(self, X):
|
||||||
return X + self.biases.f
|
return X + self.biases.f
|
||||||
|
|
Loading…
Add table
Reference in a new issue