직접 구현한 코드에 대한 리뷰 요청

너는 딥러닝 프레임워크에 정통한 시니어 엔지니어야. 아래 코드는 내가 밑바닥부터 구현해본 {{구현한 대상, 예: 셀프 어텐션 클래스}}야.

1) 텐서 차원이 각 단계에서 어떻게 바뀌는지 줄 단위로 추적해서 알려줘
2) 버그나 비효율적인 부분이 있으면 구체적인 줄 번호와 함께 지적해줘
3) 실제 유명 구현(예: PyTorch 공식 구현)과 다른 점이 있다면 알려줘

```python
{{코드 붙여넣기}}
```