Questions about the EGNN code

[Byun-jinyoung]

Recently, I've tried to read EGNN paper and study your EGNN code.
Actually, I had hard time to understand both paper and code because my major is not computer science.
When studying your code, I realize that the shape of hidden_out and the shape of kwargs["x"] must be same to perform add operation (becaus of residual connection) in the class EGNN_sparse forward method.
How can I increase or decrease the hidden dimension size of x?

I would like to get some advice.

Thanks for your consideration in this regard.