Softmax Jacobian [medium]

v1 소프트맥스 는 forward. 역전파에서 필요한 건 Jacobian $J_{ij} = \dfrac{\partial p_i}{\partial x_j}$:

$\frac{\partial p_i}{\partial x_j} = p_i (\delta_{ij} - p_j) = \begin{cases} p_i (1 - p_i) & i = j \\ -p_i p_j & i \ne j \end{cases}$

행렬 형태:

$J = \text{diag}(p) - p p^\top$

• 대각: $p_i - p_i^2 = p_i(1 - p_i)$
• 비대각: $-p_i p_j$

성질

• 대칭: $J = J^\top$.
• 행/열 합 = 0: 확률 합이 1 로 제약된 결과 — any probability conservation identity.
• 양의 준정부호: $J$ 의 eigenvalue 는 모두 $\ge 0$ — softmax가 convex potential 의 그래디언트이기 때문.

과제

함수 softmax_jacobian(x) 를 완성하세요.

• x shape (n,) 로짓.
• 반환: shape (n, n) Jacobian.
• 힌트:

  p = stable_softmax(x)          # (n,)
  return np.diag(p) - np.outer(p, p)
  

테스트 케이스