Write a Custom Activation Function in PyTorch

Basicauto-graded

Implement a custom activation function as a PyTorch module and use it inside a neural network.

Solve it

Open in ColabQuestion notebookSolution

Check your answer

The grader verifies properties of your implementation, so a correct solution written differently from ours still passes.

pip install torchleet

from torchleet import check
check("custom-activation", CustomActivationModel)