error abou model.py_forward

[Henry-Bi]

File "/home/sc/carl_test/BottleneckTransformers-main/model.py", line 43, in forward
energy = content_content + content_position
RuntimeError: The size of tensor a (4) must match the size of tensor b (196) at non-singleton dimension 1

Hi ! when run the model.py i meet this error

[leaderj1001]

Thanks for comments.

Could you plz show me your input shape?

[YePG]

RuntimeError: The size of tensor a (4) must match the size of tensor b (196) at non-singleton dimension 2

I'm using your code. And I didn't make any changes