tracel-ai / burn

Burn is a new comprehensive dynamic Deep Learning Framework built using Rust with extreme flexibility, compute efficiency and portability as its primary goals.

Home Page:https://burn.dev

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Adding burn::nn::Sigmoid

RuelYasa opened this issue · comments

Feature description

Add a Sigmoid struct in burn::nn, just like Relu and Gelu etc.

Feature motivation

Sigmoid struct doesn't exist, but nn::Relu and nn::Gelu exist. I havn't figured out why sigmoid can't be used as a struct like them, so it might be reasonable to do this.

(Optional) Suggest a Solution