我是靠谱客的博主 追寻羽毛,最近开发中收集的这篇文章主要介绍PyTorch学习笔记(17) ——pytorch的torch.repeat和tf.tile的对比,觉得挺不错的,现在分享给大家,希望可以做个参考。

概述

0. 前言

在对比stylegan2的D_stylegan2的时候, 我需要校对我的Minibatch_stddev_layer和作者的tf版本的区别,其中,有一个算子f.tile, 网上已经介绍的很清楚了,将其翻译为pytorch版本的时候其对应的算子没有清晰的解释,因此我将torch.repeattf.tile进行了简单比较(下面),基本结论是2者是一致的,可以等效替换.

1. 测试比较

① 二维

  • TF (1.14)
    在这里插入图片描述

  • PyTorch 1.1
    在这里插入图片描述

② 三维

  • TF (1.14)
    在这里插入图片描述
  • PyTorch 1.1
    在这里插入图片描述
    ③ 四维
  • TF (1.14)
    在这里插入图片描述
  • PyTorch 1.1
    在这里插入图片描述

2. 结论

由试验阶段可以看出,tf.tiletorch.repeat的使用是一样的. 在翻译项目的时候,直接替换即可.

3. 参考资料

[1] 直观的理解tensorflow中的tf.tile()函数
[2] pytorch官方文档:repeat

最后

以上就是追寻羽毛为你收集整理的PyTorch学习笔记(17) ——pytorch的torch.repeat和tf.tile的对比的全部内容,希望文章能够帮你解决PyTorch学习笔记(17) ——pytorch的torch.repeat和tf.tile的对比所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(39)

评论列表共有 0 条评论

立即
投稿
返回
顶部