torch.gather()之通俗易懂讲解

文章目录

  • 1. 简要介绍
  • 2. 深入讲解之化繁为简
  • 3. 总结

1. 简要介绍

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

torch.gather()之通俗易懂讲解_第1张图片
  torch.gather()对应官网链接为:https://pytorch.org/docs/stable/generated/torch.gather.html#torch.gather,字面含义如下图所示:

你可能感兴趣的:(机器学习入门之工具篇,pytorch,深度学习)