Flux.jl/test/batching.jl
2017-02-20 19:34:57 +00:00

8 lines
126 B
Julia

bs = Batch([[1,2,3],[4,5,6]])
@test typeof(bs) <: Batch
@test bs == [[1,2,3],[4,5,6]]
@test rawbatch(bs) == [1 2 3; 4 5 6]