
2번 의 응용. 한 번에 한 쌍이 아니라 여러 쌍의 내적을 동시에 계산합니다. 실제 ML 파이프라인에선 이게 기본:
Q·K 각 위치별 계산
는 shape (N, D) — 결과는 shape (N,).
axis=12D 배열에서 "행별로 합치기"는 axis=1. np.sum(X * Y, axis=1) 만 있으면 끝.
함수 batch_dot(X, Y) 를 완성하세요.
X, Y shape (N, D). 반환: shape (N,).| # | 이름 | 검증 |
|---|---|---|
| 1 | shape (N,) | 입력 (5, 3) → 출력 (5,) |
| 2 | 원소별 일치 | 각 i에서 result[i] == X[i] @ Y[i] |
| 3 | 직교 쌍 → 0 | 단위행렬 순열 입력 |
| 4 | 대칭 | batch_dot(X, Y) == batch_dot(Y, X) |
| 5 | 루프 없이 | for/while 금지 |
코드를 작성하고 Run 을 눌러보세요.