如何在Flux.jl中定义自定义损失函数？

Question

查看Flux.jl文档，我发现有很多内置的损失函数：https://fluxml.ai/Flux.jl/stable/models/losses/。我的问题是，如果我想要更深奥的东西，我如何在Flux中定义和使用我自己的损失函数？

Answer 1

您可以使用任何返回单个浮点值作为损失的可微函数，如上面的注释所述，准备好的函数只是为了您的方便。你可以传递任何东西，例如

using Flux
yourcustomloss(ŷ, y) = sum(.- sum(y .* logsoftmax(ŷ), dims = 1))

并计算其梯度或将其传递给train!函数。