前提・実現したいこと
PyTorchを用いてチャンネルごとに任意の異なる値でパディングしたいです.
PyTorchにはパディング用の関数torch::nn::functional::Pad
が用意されています.
この関数は引数でmode="constant", value=0
のようにするとパディングの値を変えることができます.
しかし,この関数では,私の知る限りチャンネルごとにパディングの値を変えることはできません.
チャンネルごとにパディングの値を変えるためのアドバイスをください.
よろしくお願いします.
補足情報(FW/ツールのバージョンなど)
python 3.6.5
pytorch 1.4.0
あなたの回答
tips
プレビュー