Thanks for your reply.
I have tried groups parameter, it’s a good direction.
But it cannot give me the correct output, because reshape weight cannot give the correct weight order for this scenario.
Maybe I need to redefine a reshape function to support.