Skip to content

Commit b6f3c97

Browse files
committed
update tests for linear2d layer
1 parent eff36fe commit b6f3c97

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

test/test_linear2d_layer.f90

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ program test_linear2d_layer
1212
type(linear2d_layer) :: linear
1313

1414
linear = linear2d_layer(batch_size=2, sequence_length=3, in_features=4, out_features=1)
15+
call linear % init([4])
1516

1617
call test_linear2d_layer_forward(linear, ok, sample_input)
1718
call test_linear2d_layer_backward(linear, ok, sample_input, sample_gradient)

0 commit comments

Comments
 (0)