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
最后
以上就是追寻羽毛最近收集整理的关于PyTorch学习笔记(17) ——pytorch的torch.repeat和tf.tile的对比的全部内容,更多相关PyTorch学习笔记(17)内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复