update api
This commit is contained in:
parent
6808a92793
commit
b8565a4cc3
|
@ -13,7 +13,7 @@ end
|
|||
Input(i...) = Input(dims(i...))
|
||||
|
||||
(::Input)(x) = x
|
||||
back!(::Input, ∇) = ∇
|
||||
back!(::Input, ∇, x) = ∇
|
||||
|
||||
shape(i::Input) = i.dims
|
||||
|
||||
|
|
Loading…
Reference in New Issue