From fc3af681ec45f9fb8ea405a447908c1b8bc9bbed Mon Sep 17 00:00:00 2001
From: Garben Tanghe <garben.tanghe@gmail.com>
Date: Sun, 8 Mar 2020 14:17:00 +0100
Subject: [PATCH] updated documentation

---
 src/layers/stateless.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/layers/stateless.jl b/src/layers/stateless.jl
index a9e6c6e5..eebbbe98 100644
--- a/src/layers/stateless.jl
+++ b/src/layers/stateless.jl
@@ -204,7 +204,7 @@ tversky_loss(ŷ, y; β=eltype(ŷ)(0.7)) = 1 - (sum(y .* ŷ) + 1) / (sum(y .*
 """
     flatten(x::AbstractArray)
 
-Transforms (w,h,c,b)-shaped input into (w*h*c,b)-shaped output,
+Transforms (w,h,c,b)-shaped input into (w x h x c,b)-shaped output,
 by linearizing all values for each element in the batch.
 """
 function flatten(x::AbstractArray)