When using grouped convolution, the shape of new_weight is incorrect, causing an error in the convolution calculation. <img width="554" alt="image" src="https://github.com/user-attachments/assets/9c4ed715-860b-45dd-9780-a4b57d61d9c2" />
When using grouped convolution, the shape of new_weight is incorrect, causing an error in the convolution calculation.
