re-export gradient

2019-07-11 13:55:12 +01:00 · 2019-07-11 13:55:12 +01:00 · c2cd7dab91
commit c2cd7dab91
parent cf5bc801d3
1 changed files with 2 additions and 1 deletions
--- a/src/Flux.jl
+++ b/src/Flux.jl
@ -7,6 +7,7 @@ using MacroTools, Juno, Requires, Reexport, Statistics, Random
 using MacroTools: @forward
@reexport using NNlib
 using Zygote: Params, @adjoint, gradient
+export gradient

 export Chain, Dense, Maxout, RNN, LSTM, GRU, Conv, CrossCor, ConvTranspose, MaxPool, MeanPool,
       DepthwiseConv, Dropout, AlphaDropout, LayerNorm, BatchNorm, InstanceNorm, GroupNorm,