unbatch vector

Author: CarloLucibello

src/utils.jl

@@ -583,6 +583,7 @@ julia> Flux.unbatch([1 3 5 7;
  [7, 8]
 """
 unbatch(x::AbstractArray) = unstack(x, ndims(x))
+unbatch(x::AbstractVector) = x
 
 """
 Return the given sequence padded with `p` up to a maximum length of `n`.

test/utils.jl

@@ -340,6 +340,14 @@ end
   unstacked_array=[[8, 9, 9, 7], [9, 6, 1, 4], [3, 6, 7, 10], [5, 9, 2, 6]]
   @test unbatch(stacked_array) == unstacked_array
   @test batch(unstacked_array) == stacked_array
+
+  # no-op for vector of non-arrays
+  @test batch([1,2,3]) == [1,2,3]
+  @test unbatch([1,2,3]) == [1,2,3]
+
+  # generic iterable
+  @test batch(ones(2) for i=1:3) == ones(2, 3)
+  @test unbatch(ones(2, 3)) == [ones(2) for i=1:3]
 end
 
 @testset "Param remapping" begin