can not export 'ingroup_indices' to torchscript

[cyiano]

SST model relies on function 'IngroupIndicesFunction', which needs ingroup_indices op. But it is unable to export to torchscript. How can I fix it.

Error message: Could not export Python function call 'IngroupIndicesFunction'. Remove calls to Python functions before export. Did you forget to add @script or @script_method annotation? If this is a nn.ModuleList, add it to constants:

[rockywind]

Hi， I met the same error. How can you solve it?

[Abyssaledge]

I have no experience with torchscript. However, it is a very simple op, so it should be easy to implement it in your way. @rockywind

[rockywind]

Thanks!