ONNX何时决定包括收集操作?

时间 2019-02-19
阅读 34
点赞 0
收藏 0
连接mrwheet

我想这个问题不太清楚,但我刚建立了一个pytorch模型,它无法转换为coreml,因为onnx已经添加了Gather操作系统。整个模型实际上是两个独立模型的合并,旨在通过尽可能长时间地保持对GPU/金属的处理来提高性能。

构建这个“复合”模型需要我创建两个表单切片x = y[:, 0]我想知道这些是否是Gather操作系统?

我确实意识到我可以创建一个自定义层,但是我刚刚经历了一个可怕的失败,在coreml中使用自定义层,这浪费了很多时间,而且我什么都没有,所以我正在尝试另一种解决问题的方法。

如果找到绕过这些切片的方法,将阻止onnx添加Gather我愿意寻找解决办法。

任何想法都值得赞赏。

尚无答案