Skip to content

Commit cb3167c

Browse files
committed
fix(Dense): bad variable init
1 parent 9f77e66 commit cb3167c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

neuralnetlib/layers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -149,9 +149,9 @@ def forward_pass(self, input_data: np.ndarray) -> np.ndarray:
149149

150150
if len(input_data.shape) == 1 and self.input_dim:
151151
batch_size = input_data.shape[0]
152-
input_data = input_data.reshape(batch_size, input_data = input_data.reshape(batch_size, self.input_dim))
152+
input_data = input_data.reshape(batch_size, self.input_dim)
153153
self.input = input_data
154-
154+
155155
if len(input_data.shape) == 3:
156156
batch_size, timesteps, features = input_data.shape
157157
input_reshaped = input_data.reshape(-1, features)

0 commit comments

Comments
 (0)