Merge pull request #583 from KristofferC/kc/small_fixes

clarify docs on single batch image to conv
This commit is contained in:
Mike J Innes 2019-02-04 12:33:34 +00:00 committed by GitHub
commit 17f33b4a6a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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`.
"""