basic batch type

src/Flux.jl

@@ -27,4 +27,6 @@ include("layers/stateless.jl")
 include("layers/basic.jl")
 include("layers/recurrent.jl")
 
+include("batches/Batches.jl")
+
 end # module

src/batches/Batches.jl

@@ -0,0 +1,7 @@
+module Batches
+
+import ..Flux
+
+include("batch.jl")
+
+end

src/batches/batch.jl

@@ -0,0 +1,8 @@
+struct Batch{T,A,M}
+  data::A
+  mask::M
+end
+
+Batch{T}(data, mask) where T = Batch{T,typeof(data),typeof(mask)}(data, mask)
+
+Batch(xs) = Batch{typeof(first(xs))}(Flux.batch(xs),trues(length(xs)))