Torch.onnx.export alternate layers for SnapML importing?

I’ve implemented the SINet model from this paper so that I can export it to a .onnx and then to SnapML.

However SnapML will not import some onnx layers like “Gather” which are generated from torch.onnx.export() for common methods like torch.tensor.view(). Is there an equivalent to torch.tensor.view() that torch.onnx won’t export to a “Gather” layer?