PyTorch學習筆記(17) ——pytorch的torch.repeat和tf.tile的對比

0. 前言

在對比stylegan2的D_stylegan2的時候, 我需要校對我的Minibatch_stddev_layer和作者的tf版本的區別，其中，有一個算子f.tile, 網上已經介紹的很清楚了，將其翻譯爲pytorch版本的時候其對應的算子沒有清晰的解釋，因此我將torch.repeat和tf.tile進行了簡單比較(下面)，基本結論是2者是一致的，可以等效替換.

1. 測試比較

① 二維

• TF （1.14）
  

• PyTorch 1.1
  

② 三維

• TF （1.14）
  
• PyTorch 1.1
  
  ③ 四維
• TF （1.14）
  
• PyTorch 1.1
  

2. 結論

由試驗階段可以看出，tf.tile和torch.repeat的使用是一樣的. 在翻譯項目的時候，直接替換即可.

3. 參考資料

[1] 直觀的理解tensorflow中的tf.tile()函數
[2] pytorch官方文檔：repeat