Add soft plus layer for torch

lanl-ansi / MathOptAI.jl

Embed trained machine learning predictors in JuMP

https://lanl-ansi.github.io/MathOptAI.jl/

Other

29 stars 1 forks source link

Closed odow closed 1 month ago

odow commented 1 month ago

elseif Bool(PythonCall.pybuiltins.isinstance(layer, nn.Softplus))
        return get(config, :SoftPlus, MathOptAI.SoftPlus())