GPU版本报错

[Hongyuan-Liu]

你好，我跑你的实现，CPU版本没有问题，但是设置为GPU模式时报下面的错误：
2024-03-19 08:12:34.290537612 [E:onnxruntime:, sequential_executor.cc:514 ExecuteKernel] Non-zero status code returned while running Expand node. Name:'/class_embed.0_4/Expand' Status Message: /class_embed.0_4/Expand: left operand cannot broadcast on dim 2 LeftShape: {1,900,8}, RightShape: {1,900,256}
terminate called after throwing an instance of 'Ort::Exception'
what(): Non-zero status code returned while running Expand node. Name:'/class_embed.0_4/Expand' Status Message: /class_embed.0_4/Expand: left operand cannot broadcast on dim 2 LeftShape: {1,900,8}, RightShape: {1,900,256}