MCPcopy Create free account
hub / github.com/pytorch/examples / ActivationFunction

Class ActivationFunction

fx/wrap_output_dynamically.py:33–36  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

31
32# Selected activation functions
33class ActivationFunction(Enum):
34 RELU = auto()
35 LEAKY_RELU = auto()
36 PRELU = auto()
37
38# Map activation function names to their implementation
39activation_functions = {

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected