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

comment fix

Browse Source
This commit is contained in:
Dhairya Gandhi 2019-02-03 04:00:08 +05:30
parent bd6158d7f9
commit e243950e28
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/onehot.jl
View File

@ -28,7 +28,7 @@ Base.getindex(xs::Flux.OneHotMatrix, j::Base.UnitRange, i::Int) = xs.data[i][j]
Base.getindex(xs::OneHotMatrix, ::Colon, ::Colon) = xs
# handle special case for when we want the entire column without allocating
# handle special case for when we want the entire column
function Base.getindex(xs::Flux.OneHotMatrix{T}, ot::Union{Base.Slice, Base.OneTo}, i::Int) where {T<:AbstractArray}
res = similar(xs, size(xs, 1), 1)
if length(ot) == size(xs, 1)