Skip to content
This repository has been archived by the owner on Mar 26, 2019. It is now read-only.

ONNX Operator "Gather" is not implemented by onnx_mxnet #24

Open
HeliWang opened this issue Feb 19, 2018 · 2 comments
Open

ONNX Operator "Gather" is not implemented by onnx_mxnet #24

HeliWang opened this issue Feb 19, 2018 · 2 comments

Comments

@HeliWang
Copy link

In https://github.com/onnx/onnx-mxnet/blob/0ca9a87088619499c932a78cc2871a14b87ab20d/onnx_mxnet/import_helper.py, "Gather" operator is commented. Any idea of solution when the ONNX graph uses “Gather” operator? Thanks

@Roshrini
Copy link
Collaborator

ONNX's "Gather" operator can be mapped directly to MXNet's "take" operator.
Exception: Currently, "take" only supports axis 0.

You are welcome to contribute. You can do so by adding a mapping here:
https://github.com/onnx/onnx-mxnet/blob/master/onnx_mxnet/import_helper.py#L217
Or I will add a fix soon.

@HeliWang
Copy link
Author

Pull request has been made: rajanksin/incubator-mxnet#1
@spidydev @Roshrini could you help to have a look? Thanks.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

No branches or pull requests

2 participants