Skip to content

Commit 30944a9

Browse files
Update conv.jl
Specifying the format of the data given to the convolutional layer.
1 parent c99be6c commit 30944a9

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/layers/conv.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,8 @@ Image data should be stored in WHCN order (width, height, channels, batch).
4040
In other words, a 100×100 RGB image would be a `100×100×3×1` array,
4141
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)`,
43-
a 2-tuple of integers.
43+
a 2-tuple of integers. For different `N`, data should be `N_1 x N_2 x ... x N_n x channels x batch`,
44+
i.e. `spatial/temporal dimensions x channels x batch`.
4445
4546
For `N` spatial dimensions, this layer expects as input an array
4647
with `ndims(x) == N+2`, where `size(x,N+1) == in` is the number of channels.

0 commit comments

Comments
 (0)