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

1.0+ updates - indices to axes, Vector init with undef

Browse Source
This commit is contained in:
Robert Luciani 2018-10-02 21:39:00 +02:00
parent 02ecca4c61
commit 252e34e173
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/utils.jl
View File

@ -24,7 +24,7 @@ julia> chunk(1:10, 3)
""" """
chunk(xs, n) = collect(Iterators.partition(xs, ceil(Int, length(xs)/n))) chunk(xs, n) = collect(Iterators.partition(xs, ceil(Int, length(xs)/n)))
batchindex(xs, i) = (reverse(Base.tail(reverse(indices(xs))))..., i) batchindex(xs, i) = (reverse(Base.tail(reverse(axes(xs))))..., i)
""" """
frequencies(xs) frequencies(xs)
@ -66,7 +66,7 @@ julia> batch([[1,2,3],[4,5,6]])
function batch(xs) function batch(xs)
data = first(xs) isa AbstractArray ? data = first(xs) isa AbstractArray ?
similar(first(xs), size(first(xs))..., length(xs)) : similar(first(xs), size(first(xs))..., length(xs)) :
Vector{eltype(xs)}(length(xs)) Vector{eltype(xs)}(undef, length(xs))
for (i, x) in enumerate(xs) for (i, x) in enumerate(xs)
data[batchindex(data, i)...] = x data[batchindex(data, i)...] = x
end end