Skip to content

Commit fcda965

Browse files
committed
Merge branch 'master' of https://github.com/racinmat/Flux.jl
2 parents 0a67472 + f221ee9 commit fcda965

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/onehot.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -235,7 +235,7 @@ end
235235
function Base.:(*)(A::AbstractMatrix, B::Adjoint{Bool, <:OneHotMatrix})
236236
B_dim = length(parent(B).indices)
237237
size(A, 2) == B_dim || throw(DimensionMismatch("Matrix column must correspond with OneHot size: $(size(A, 2)) != $B_dim"))
238-
return NNlib.scatter(+, A, parent(B).indices, dstsize=(size(A,1), size(B,2)))
238+
return NNlib.scatter(+, A, _indices(parent(B)), dstsize=(size(A,1), size(B,2)))
239239
end
240240

241241
for wrapper in [:Adjoint, :Transpose]

0 commit comments

Comments
 (0)