当前位置:   article > 正文

torch.gather()之通俗易懂讲解_paddle转pytorch gathernd

paddle转pytorch gathernd

1. 简要介绍

  通过在GitHub网站上检索可得,torch.gather()在4W多的代码文件中得到了使用,所以可见它是Pytorch中个非常常用的函数。如下图所示:

在这里插入图片描述
  torch.gather()对应官网链接为:https://pytorch.org/docs/stable/generated/torch.gather.html#torch.gather,字面含义如下图所示:
在这里插入图片描述

  其中最核心的公式如下所示,但是它看起来非常不友好,我第一次学习的时候也感到非常的困惑!!!

out[i][j][k]
    声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小舞很执着/article/detail/897329
    推荐阅读