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?