← 返回 amazon 的题目列表Implement GPT Model
类型：online_judge
Implement a basic GPT model including the following classes: MultiheadAttention, Feedforward, DecoderLayer, and GPT. Each class should be able to handle the respective input and output tensors. Ensure that the model can perform forward propagation operations.
Example
Input
2
1000
256
6
8
0.1
100