Skip to content

Commit 726bbdd

Browse files
Merge #1646
1646: Update conv.jl r=mcabbott a=vivekkumar7089 Specifying the format of the data given to the convolutional layer, as mentioned in #1465 . Co-authored-by: Vivek Kumar <60113444+vivekkumar7089@users.noreply.github.com>
2 parents 9410677 + bddb5aa commit 726bbdd

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/layers/conv.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,9 @@ and a batch of 50 would be a `100×100×3×50` array.
4242
This has `N=2` spatial dimensions, and needs a kernel size like `(5,5)`,
4343
a 2-tuple of integers.
4444
45-
For `N` spatial dimensions, this layer expects as input an array
46-
with `ndims(x) == N+2`, where `size(x,N+1) == in` is the number of channels.
45+
To take convolutions along `N` feature dimensions, this layer expects as input an array
46+
with `ndims(x) == N+2`, where `size(x, N+1) == in` is the number of input channels,
47+
and `size(x, ndims(x))` is (as always) the number of observations in a batch.
4748
Then:
4849
* `filter` should be a tuple of `N` integers.
4950
* Keywords `stride` and `dilation` should each be either single integer,

0 commit comments

Comments
 (0)