1 Repositories
Python pytorch_remove_ScatterND Libraries
pytorchのスライス代入操作をonnxに変換する際にScatterNDならないようにするサンプル
pytorch_remove_ScatterND pytorchのスライス代入操作をonnxに変換する際にScatterNDならないようにするサンプル。 スライスしたtensorにそのまま代入してしまうとScatterNDになるため、計算結果をcatで新しいtensorにする。 python ver