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

docs fix

This commit is contained in:
Dhairya Gandhi 2020-02-26 22:41:45 +05:30
parent cf82393ae8
commit 20e78e274e
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/layers/conv.jl
View File

@ -7,6 +7,7 @@ _convtransoutdims(isize, ksize, ssize, dsize, pad) = (isize .- 1).*ssize .+ 1 .+
expand(N, i::Tuple) = i expand(N, i::Tuple) = i
expand(N, i::Integer) = ntuple(_ -> i, N) expand(N, i::Integer) = ntuple(_ -> i, N)
""" """
Conv(filter::Tuple, in=>out) Conv(filter::Tuple, in=>out)
Conv(filter::Tuple, in=>out, activation) Conv(filter::Tuple, in=>out, activation)
@ -127,7 +128,7 @@ outdims(l::Conv, isize) =
""" """
ConvTranspose(size, in=>out) ConvTranspose(size, in=>out)
ConvTranspose(size, in=>out, relu) ConvTranspose(size, in=>out, activation)
Standard convolutional transpose layer. `filter` should be a tuple like `(2, 2)`. Standard convolutional transpose layer. `filter` should be a tuple like `(2, 2)`.
`in` and `out` specify the number of input and output channels respectively. `in` and `out` specify the number of input and output channels respectively.