introduce Reexport

- Reexporting NNlib

fix #180
This commit is contained in:
Iblis Lin 2018-02-21 16:55:20 +08:00
parent 4035745f6e
commit 043fedde3c
No known key found for this signature in database
GPG Key ID: 003085025A39EA77
2 changed files with 3 additions and 4 deletions

View File

@ -9,6 +9,7 @@ GZip
Colors
ZipFile
AbstractTrees
Reexport
# AD
ForwardDiff 0.5.0

View File

@ -4,7 +4,7 @@ module Flux
# Zero Flux Given
using Juno, Requires
using Juno, Requires, Reexport
using MacroTools: @forward
export Chain, Dense, RNN, LSTM, GRU, Conv2D,
@ -12,9 +12,7 @@ export Chain, Dense, RNN, LSTM, GRU, Conv2D,
SGD, ADAM, Momentum, Nesterov, AMSGrad,
param, params, mapleaves
using NNlib
export σ, sigmoid, logσ, logsigmoid, relu, leakyrelu, elu, swish, softmax, logsoftmax,
conv2d, maxpool2d, avgpool2d
@reexport using NNlib
include("tracker/Tracker.jl")
using .Tracker