fix: Getting biases

castelao · castelao · commit abdb7dcc6436 · 2024-06-23T12:49:04.000-06:00
diff --git a/src/nf/nf_rnn_layer_submodule.f90 b/src/nf/nf_rnn_layer_submodule.f90
@@ -114,8 +114,9 @@ module subroutine set_params(self, params)
 
     ! reshape the biases
     first = last + 1
+    last = first + self % output_size
     self % biases = reshape( &
-      params(first:), &
+      params(first:last), &
       [self % output_size] &
     )
 

Original file line number	Diff line number	Diff line change
`@@ -114,8 +114,9 @@ module subroutine set_params(self, params)`
`114`	`114`
`115`	`115`	`! reshape the biases`
`116`	`116`	`first = last + 1`
	`117`	`+ last = first + self % output_size`
`117`	`118`	`self % biases = reshape( &`
`118`		`- params(first:), &`
	`119`	`+ params(first:last), &`
`119`	`120`	`[self % output_size] &`
`120`	`121`	`)`
`121`	`122`