TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

introduce Reexport

Browse Source 
- 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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
REQUIRE
View File

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

6
src/Flux.jl
View File

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