Bug Report for transformer-block

[Voldemort108X]

Bug Report for https://neetcode.io/problems/transformer-block

Please describe the bug below and include any steps to reproduce the bug or screenshots if possible.

The solution code returns

[[[-2.1971,-2.1731,-1.1719,-0.8711],[0.6966,0.3153,-0.2754,-2.3385]],[[-0.5898,-2.4014,-0.3871,0.3969],[-0.5150,0.4576,0.9430,-0.4617]]]

which is inconsistent with the expected output

[[[-2.3930,-2.1731,-1.3218,-0.4108],[0.6966,0.3153,-0.2754,-2.3385]],[[-0.5898,-2.4014,-0.3871,0.3969],[-0.5150,0.4576,0.9430,-0.4617]]]