自定义博客皮肤VIP专享

*博客头图：

点击选择上传的图片

格式为PNG、JPG，宽度*高度大于1920*100像素，不超过2MB，主视觉建议放在右侧，请参照线上博客头图

请上传大于1920*100像素的图片！

博客底图：

点击选择上传的图片

图片格式为PNG、JPG，不超过1MB，可上下左右平铺至整个背景

栏目图：

点击选择上传的图片

图片格式为PNG、JPG，图片宽度*高度为300*38像素，不超过0.5MB

主标题颜色：

RGB颜色，例如：#AFAFAF

Hover：

RGB颜色，例如：#AFAFAF

副标题颜色：

RGB颜色，例如：#AFAFAF

预览取消提交

自定义博客皮肤

-+

上一步保存

weixin_42200930的博客

torch

关注

文章平均质量分 79

关注数：文章数：1 文章阅读量：20785 文章收藏量：130

作者: 今天不用睡觉了、

在校生，对深度学习有一定的兴趣

展开

pytorch函数（1）：torch.gather()的正确理解方法

torch.gather()pytorch 官方文档torch.gather(input, dim, index, *, sparse_grad=False, out=None) → Tensor作用：沿着由dim指定的轴收集数值。参数input (Tensor) – 目标变量，输入dim (int) – 需要沿着取值的坐标轴index (LongTensor) – 需要取值的索引矩阵sparse_grad (bool,optional) – 如果为真，输入将是一个稀疏张量out (

原创 2020-10-10 15:20:33 · 20788 阅读 · 11 评论