We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a99d800 commit ea0012aCopy full SHA for ea0012a
src/nf/nf_layer_constructors_submodule.f90
@@ -69,6 +69,8 @@ end function dense
69
module function dropout(rate) result(res)
70
real, intent(in) :: rate
71
type(layer) :: res
72
+ if (rate < 0 .or. rate > 1) &
73
+ error stop 'rate must be between 0 and 1 in a dropout layer'
74
res % name = 'dropout'
75
allocate(res % p, source=dropout_layer(rate))
76
end function dropout
@@ -81,7 +83,6 @@ module function flatten() result(res)
81
83
end function flatten
82
84
85
-
86
module function input1d(layer_size) result(res)
87
integer, intent(in) :: layer_size
88
0 commit comments