Pls Help Assertion error get_angles

Dear mentor,

Im getting below error while implementing code for get_angle function.

How it is possible to get array of shape ({4},{16}). How its possible

assert result.shape == (position, d_model), f"Wrong shape. We expected: ({position}, {d_model})"

Please click my name and message your notebook as an attachment.

Sir sent u notebook, kindly pls check

passed the test case sir thankā€¦ i didnt apply the power function