使用 NumPy 写出分段函数？

Question

我是 NumPy 的新手，并试图弄清楚如何写下分段定义的 function。 我有一个 MSE_Loss function 的衍生物，使用 ReLU 作为激活 function：

我有我需要的所有参数。 我的问题是：如何使用 NumPy 来处理分段函数（最好不使用循环）？

已编辑：例如，我可以使用 NumPy 以下方式写下以下 function ：

def loss_derivative(X, y, w, b):
    n = len(y)
    sigma = sigmoid(X @ w + b)
    return (X.T @ ((sigma - y) * sigma * (1 - sigma))) / n

如何使用上面的分段公式相同的输入来实现 function？

Answer 1

我认为此示例最简单的方法是使用条件乘数，如下所示（并非 function 中的所有元素都在您的问题中定义，所以我猜测它们。但是，如果不是这种情况，您可以轻松修复它）：

def loss_derivative(X, y, w):
    n = len(y)
    return X.T @ (((X @ w - y) / n) * ((X @ w) > 0))