convertel for batch

2017-01-26 23:37:06 +05:30 · 2017-01-26 23:37:06 +05:30 · b79e536c13
commit b79e536c13
parent 8db503eafa
1 changed files with 4 additions and 0 deletions
--- a/src/dims/batching.jl
+++ b/src/dims/batching.jl
@ -36,3 +36,7 @@ function rebatch(xs)
  B = Array{eltype(xs),dims+1}
  Batch{T,B}(xs)
 end
+
+convertel(T::Type, xs::Batch) =
+  isa(eltype(eltype(xs)), T) ? xs :
+    Batch(map(x->convertel(T, x), xs))