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