this should be a general util

2017-05-01 13:45:24 +01:00 · 2017-05-01 13:45:24 +01:00 · d095c640c9
commit d095c640c9
parent eae13c533f
1 changed files with 4 additions and 1 deletions
--- a/src/dims/batching.jl
+++ b/src/dims/batching.jl
@ -1,4 +1,4 @@
-export Batch, batchone
+export Batch, batchone, tobatch

 struct Batch{T,S} <: AbstractVector{T}
  data::CatMat{T,S}
@ -30,3 +30,6 @@ convertel(T::Type, xs::Batch) =

 batchone(x) = Batch((x,))
 batchone(x::Batch) = x
+
+tobatch(xs::Batch) = rawbatch(xs)
+tobatch(xs) = tobatch(batchone(xs))