やりたかったこと PyTorchモデルをONNXファイルにexportしたい モデルにはカスタムオペレータを含む カスタムオペレータの実装は存在しない（該当部分は推論不能） カスタムオペレータはONNXでそれらしきop_typeを持つノードで出力してくれればOK やってみた…

Axial-DeepLab (ECCV 2020, Spotlight、arXiv:2003.07853)の論文1を読んでいて、Axial-Attentionが具体的にどんなものなのかよく分からなかったので、色々調べたり著者実装2を自分で実行してみたりして「こんな感じかなぁ」という認識に至ったので内容をメモ…