TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

this should be a general util

Browse Source
This commit is contained in:
Mike J Innes 2017-05-01 13:45:24 +01:00
parent eae13c533f
commit d095c640c9
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/dims/batching.jl
View File

@ -1,4 +1,4 @@
export Batch, batchone export Batch, batchone, tobatch
struct Batch{T,S} <: AbstractVector{T} struct Batch{T,S} <: AbstractVector{T}
data::CatMat{T,S} data::CatMat{T,S}
@ -30,3 +30,6 @@ convertel(T::Type, xs::Batch) =
batchone(x) = Batch((x,)) batchone(x) = Batch((x,))
batchone(x::Batch) = x batchone(x::Batch) = x
tobatch(xs::Batch) = rawbatch(xs)
tobatch(xs) = tobatch(batchone(xs))