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