File tree 1 file changed +5
-10
lines changed
1 file changed +5
-10
lines changed Original file line number Diff line number Diff line change 1
1
"""
2
- Upsample(mode = :nearest; scale = nothing , size = nothing )
2
+ Upsample(mode = :nearest; [ scale, size] )
3
3
4
- An upsampling layer.
4
+ An upsampling layer. One of two keywords must be given:
5
5
6
- `scale` is a number or a tuple of numbers
7
- representing the output rescaling factor along each spatial dimension.
8
- For integer `scale`, all but the last 2 dimensions (channel and batch)
9
- will be rescaled by the same factor.
10
-
11
- It is also possible to directly specify the output spatial `size`,
12
- as an alternative to using `scale`.
6
+ If `scale` is a number, this applies to all but the last two dimensions (channel and batch) of the input.
7
+ It may also be a tuple, to control dimensions individually. Alternatively, keyword
8
+ `size` accepts a tuple, to directly specify the leading dimensions of the output.
13
9
14
10
Currently supported upsampling `mode`s
15
11
and corresponding NNlib's methods are:
@@ -78,4 +74,3 @@ struct PixelShuffle
78
74
end
79
75
80
76
(m:: PixelShuffle )(x) = NNlib. pixel_shuffle (x, m. r)
81
-
You can’t perform that action at this time.
0 commit comments