From 325e3a4f70f3d4527962c3e353a0030bd8c26cf4 Mon Sep 17 00:00:00 2001 From: Kristoffer Carlsson Date: Thu, 24 Jan 2019 11:24:10 +0100 Subject: [PATCH] clarify docs on single batch image to conv fixes #309 --- src/layers/conv.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/layers/conv.jl b/src/layers/conv.jl index 99fc16f2..30ce7288 100644 --- a/src/layers/conv.jl +++ b/src/layers/conv.jl @@ -13,7 +13,7 @@ Standard convolutional layer. `size` should be a tuple like `(2, 2)`. `in` and `out` specify the number of input and output channels respectively. Data should be stored in WHCN order. In other words, a 100×100 RGB image would -be a `100×100×3` array, and a batch of 50 would be a `100×100×3×50` array. +be a `100×100×3×1` array, and a batch of 50 would be a `100×100×3×50` array. Takes the keyword arguments `pad`, `stride` and `dilation`. """