Encoder-only, Decoder-only and Encoder-Decoder models — Coding Problem 2

Implement one core operation from Encoder-only, Decoder-only and Encoder-Decoder models as a reusable Python function and test it with three cases.