使用 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))