Skip to content

Commit bde63ae

Browse files
added note specifying the last dimension as batch.
1 parent 30944a9 commit bde63ae

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/layers/conv.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ 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)`,
4343
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`.
44+
i.e. `spatial/temporal dimensions x channels x batch` (notice that the batch is the last dimension).
4545
4646
For `N` spatial dimensions, this layer expects as input an array
4747
with `ndims(x) == N+2`, where `size(x,N+1) == in` is the number of channels.

0 commit comments

Comments
 (0)