Tensorflowでpytorchのreflectionpaddingを実現する - Qiita

詰まったので備忘録です。 まだ動かしていないため、もし間違いなどがあれば教えてくださると嬉しいです。 pytorchにおいて、ReflectionPadding2Dは以下のような挙動をします。 詳しいことについては公式ドキュメントを見るとわかると思います。 >>> m = nn.ReflectionPad2d(2) >>> input = torch.arange(9, dtype=torch.float).reshape(1, 1, 3, 3) >>> input tensor([[[[0., 1., 2.], [3., 4., 5.], [6., 7., 8.]]]]) >>> m(input) tensor([[[[8., 7., 6., 7., 8., 7., 6.], [5., 4., 3., 4., 5., 4., 3.], [2., 1., 0., 1., 2., 1., 0