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

simplify _getobs

This commit is contained in:
cossio 2020-06-12 12:05:39 +02:00
parent 75692161a7
commit cb34bb848b
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/data/dataloader.jl
View File

@ -104,10 +104,7 @@ function _nobs(data::Union{Tuple, NamedTuple})
return n return n
end end
function _getobs(data::AbstractArray{T,N}, i) where {T,N} _getobs(data::AbstractArray, i) = selectdim(data, ndims(data), i)
getindex(data, ntuple(i->Colon(), Val(N-1))..., i)
end
_getobs(data::Union{Tuple, NamedTuple}, i) = map(x -> _getobs(x, i), data) _getobs(data::Union{Tuple, NamedTuple}, i) = map(x -> _getobs(x, i), data)
Base.eltype(d::DataLoader{D}) where D = D Base.eltype(d::DataLoader{D}) where D = D