python - 从枚举中获取所有值，当值在 Python 3.7 中是可调用的

Question

我有一个 Enum Python 3.7 中的类定义如下:

# activation functions
def relu(x: float) -> float:
    return x * (x > 0)


def sigmoid(x: float) -> float:
    return 1 / (1 + math.exp(-x))


class Activation(Enum):
    SIGMOID = sigmoid
    RELU = relu

    def __call__(self, x):
        return self.value(x)

    @classmethod
    def values(cls):
        return [function.value for function in cls]

我在网上尝试了其他类似问题的一些方法，例如 list(Activation)和 Activation.values()但总是得到一个空列表。有什么新想法吗？

Answer 1

如果您使用 aenum 1 个库，您可以使用它的 member描述符:

from aenum import Enum, member

class Activation(Enum):

    SIGMOID = member(sigmoid)
    RELU = member(relu)

    def __call__(self, x):
        return self.value(x)

    @classmethod
    def values(cls):
        return [function.value for function in cls]

也可以写成:

class Activation(Enum):

    @member
    def SIGMOID(x: float) -> float:
        return 1 / (1 + math.exp(-x))

    @member
    def RELU(x: float) -> float:
        return x * (x > 0)

    def __call__(self, x):
        return self.value(x)

    @classmethod
    def values(cls):
        return [function.value for function in cls]

1 披露:我是 Python stdlib Enum 的作者, enum34 backport ，以及 Advanced Enumeration ( aenum )图书馆。