adding suggested changes

Co-authored-by: Michael Abbott <32575566+mcabbott@users.noreply.github.com>

Author: vivekkumar7089

src/layers/conv.jl

@@ -43,8 +43,8 @@ This has `N=2` spatial dimensions, and needs a kernel size like `(5,5)`,
 a 2-tuple of integers.
 
 To take convolutions along `N` dimensions, this layer expects as input an array
-with `ndims(x) == N+2`, where `size(x,N+1) == in` is the number of input channels,
-and the last dimension is `size(x,N+2) == batches`.
+with `ndims(x) == N+2`, where `size(x, N+1) == in` is the number of input channels,
+and `size(x, ndims(x))` is (as always) the number of observations in a batch.
 Then:
 * `filter` should be a tuple of `N` integers.
 * Keywords `stride` and `dilation` should each be either single integer,