Update src/layers/stateless.jl

Co-authored-by: cossio <cossio@users.noreply.github.com>
This commit is contained in:
Carlo Lucibello 2020-05-05 16:42:06 +02:00 committed by GitHub
parent 79391beca0
commit 2290ced09a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -76,7 +76,7 @@ end
"""
logitcrossentropy(, y; weight=nothing, agg=mean, dims=1)
Return the cross[1.0 0.5 0.3 2.4]entropy computed after a [`Flux.logsoftmax`](@ref) operation;
Return the crossentropy computed after a [`Flux.logsoftmax`](@ref) operation;
calculated as
agg(.-sum(weight .* y .* logsoftmax(; dims=dims); dims=dims))