Obtenez un rembourrage de réflexion Pytorch avec Tensorflow

C'est un mémorandum parce qu'il est bloqué. Je ne l'ai pas encore proposé, donc si vous avez des erreurs, faites-le moi savoir.

Dans pytorch, ReflectionPadding2D se comporte comme suit. Vous pouvez en savoir plus dans la documentation officielle .

>>> 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.],
          [5., 4., 3., 4., 5., 4., 3.],
          [8., 7., 6., 7., 8., 7., 6.],
          [5., 4., 3., 4., 5., 4., 3.],
          [2., 1., 0., 1., 2., 1., 0.]]]])
>>> # using different paddings for different sides
>>> m = nn.ReflectionPad2d((1, 1, 2, 0))
>>> m(input)
tensor([[[[7., 6., 7., 8., 7.],
          [4., 3., 4., 5., 4.],
          [1., 0., 1., 2., 1.],
          [4., 3., 4., 5., 4.],
          [7., 6., 7., 8., 7.]]]])

Documentation officielle de pytorch https://pytorch.org/docs/stable/nn.html

Si vous essayez d'y parvenir avec tensorflow, vous utiliserez le pad tensorflow. (Il est écrit dans la documentation officielle ...) J'ai écrit un article parce que je ne pouvais pas le sortir de mon propre chemin.

tf.pad(
    tensor,
    paddings,
    mode='REFLECT',
    constant_values=0,
    name=None
)

Un exemple est le suivant.

t = tf.constant([[1, 2, 3], [4, 5, 6]])
paddings = tf.constant([[1, 1,], [2, 2]])
tf.pad(t, paddings, "REFLECT")  
# [[6, 5, 4, 5, 6, 5, 4],
#  [3, 2, 1, 2, 3, 2, 1],
#  [6, 5, 4, 5, 6, 5, 4],
#  [3, 2, 1, 2, 3, 2, 1]]

D'après les deux exemples de document officiel ci-dessus https://www.tensorflow.org/api_docs/python/tf/pad

Recommended Posts

Obtenez un rembourrage de réflexion Pytorch avec Tensorflow
Jouez avec PyTorch
Validation croisée avec PyTorch
À partir de PyTorch
Zundokokiyoshi avec TensorFlow
Casser des blocs avec Tensorflow
Utilisez RTX 3090 avec PyTorch
Lecture de données avec TensorFlow
Prévisions de courses de bateaux avec TensorFlow
Installer la diffusion de la torche avec PyTorch 1.7
Essayez la régression avec TensorFlow
Traduire Premiers pas avec TensorFlow
Essayez l'apprentissage en profondeur avec TensorFlow
Utiliser TensorFlow avec Intellij IDEA
Fonction sinueuse approximative avec TensorFlow
Essayez Auto Encoder avec Pytorch
Essayez d'implémenter XOR avec PyTorch
Implémenter le GPU PyTorch + avec Docker
Prévision du cours de l'action avec tensorflow
Prédiction de la moyenne Nikkei avec Pytorch 2
Démineur d'apprentissage automatique avec PyTorch
Prédiction de la moyenne Nikkei avec Pytorch
Effectuer un fractionnement stratifié avec PyTorch
J'ai créé Word2Vec avec Pytorch
Obtenez "Bals" avec Amazon Echo