search

dkurt / openvino_pytorch_layers

How to export PyTorch models with unsupported layers to ONNX and then to Intel OpenVINO
https://github.com/openvinotoolkit/openvino
Apache License 2.0
26 stars 13 forks source link

add dim param #31

Closed likholat closed 2 years ago

likholat commented 2 years ago

@dkurt, please, support dim=(1, 2) case

dkurt commented 2 years ago

see https://github.com/dkurt/openvino_pytorch_layers/pull/33