Problem when I convert Pytorch to ONNX

Hi all

I have a question concerning when I convert Pytorch to ONNX.

When I use BatchNorm2d in my model for example BatchNorm2d(32) then I convert to ONNX then I test with javascript I get not error.
Howerver when I use nn.GroupNorm in my model for example nn.GroupNorm(2,32) then I convert to ONNX then I test with javascript (I launched the chrome web browser then I check when I select “Inspecter”) I get the following error:

Uncaught (in promise) TypeError: cannot resolve operator ‘Shape’ with opsets: ai.onnx v9

Can someone tell me please how to solve this problem. The problem is when I use nn.GroupNorm.

Thanks a lot in advance

Hi all,

Does anyone have an idea how to solve this problem.

I have another question: What is the difference between ONNX, Torchscript and Torchserver?

Thanks a lot for your help.