解析Pytorch中的torch.gather()函数

 更新时间:2021-11-14 09:45:54   作者:佚名   我要评论(0)

参数说明
以官方说明为例,gather()函数需要三个参数,输入input,维度dim,以及索引index
input必须为Tensor类型
dim为int类型,代表从哪个

参数说明

以官方说明为例,gather()函数需要三个参数,输入input,维度dim,以及索引index

input必须为Tensor类型

dim为int类型,代表从哪个维度进行索引

index为LongTensor类型

举例说明

input=torch.tensor([[1,2,3],[4,5,6]]) #作为输入
 
index1=torch.tensor([[0,1,1],[0,1,1]]) #作为索引矩阵
 
# dim=0时,按列进行索引
print (torch.gather(input,dim=0,index=index1))
 
# dim=1时,按行进行索引
print (torch.gather(input,dim=1,index=index1))

 结果如下图所示:

# 按列进行索引
tensor([[1, 5, 6],
        [4, 2, 6]])
 
# 按行进行索引
tensor([[1, 2, 2],
        [5, 4, 5]])

画图说明 

官方文档

def gather(self, input, dim, index, *args, **kwargs): 
        
        For a 3-D tensor the output is specified by::
        
            out[i][j][k] = input[index[i][j][k]][j][k]  # if dim == 0
            out[i][j][k] = input[i][index[i][j][k]][k]  # if dim == 1
            out[i][j][k] = input[i][j][index[i][j][k]]  # if dim == 2        
 
        Args:
            input (Tensor): the source tensor
            dim (int): the axis along which to index
            index (LongTensor): the indices of elements to gather     
      
        Example::
        
            >>> t = torch.tensor([[1, 2], [3, 4]])
            >>> torch.gather(t, 1, torch.tensor([[0, 0], [1, 0]]))
            tensor([[ 1,  1],
                    [ 4,  3]])

到此这篇关于Pytorch中的torch.gather()函数的文章就介绍到这了,更多相关Pytorch torch.gather()函数内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
  • 浅谈Pytorch中的torch.gather函数的含义

相关文章

  • 解析Pytorch中的torch.gather()函数

    解析Pytorch中的torch.gather()函数

    参数说明 以官方说明为例,gather()函数需要三个参数,输入input,维度dim,以及索引index input必须为Tensor类型 dim为int类型,代表从哪个
    2021-11-14
  • 93充100,电费充值系统怎么做到的

    93充100,电费充值系统怎么做到的

    CPS的本质就是销售;你卖出货就赚钱。 在生活中话费水费电费充值是刚需,基本上每个月都有需求。电费充值不同于话费,电费的市场更大。
    2021-11-09
  • 月入过万 外卖小程序你知道怎么做吗

    月入过万 外卖小程序你知道怎么做吗

    怎么赚钱 美团外卖饿了么外卖赚钱小程序还能赚钱吗,老有人问能不能赚钱,那么今天就说一下逻辑 首先,分为cps悬赏和,赏金两种方式赏金属于打赏比较高的。cps的模式主要靠用户下单的结算金额,用户买的多你就赚的多,通常是在3~6%,美团是3,美团被制裁之后就降价了(可能是这个原因,据说预算不够了)。
    2021-11-08
  • 外卖CPS返利小程序源码分享

    外卖CPS返利小程序源码分享

    怎么赚钱         美团外卖饿了么外卖赚钱小程序还能赚钱吗,老有人问能不能赚钱,那么今天就说一下逻辑 首先,分为cps悬赏和,赏金两种方式赏金属于打赏比较高的。cps的模式主要靠用户下单的结算金额,用户买的多你就赚的多,通常是在3~6%,美团是3,美团被制裁之后就降价了(可能是这个原因,据说预算不够了)。  当然了,依赖外卖cps赚钱是不行的,当然要加点东西了。比如涵盖话费、视频会员、加油卡、礼品卡等多种品类折扣权益,吃、喝、玩、乐、购
    2021-11-08
  • 淘客从简单发单到精推社群,单群收益5000+的经验分享

    淘客从简单发单到精推社群,单群收益5000+的经验分享

    作为淘宝客,看到下面的这个聊天记录,应该是有点羡慕的。社群里多几个这样的粉丝,可能自己的收入也蹭蹭蹭上涨了。今天,我们就为大家邀请了运营这个社群的淘宝客–淘宝联
    2021-11-06
  • 怎样申请抖音团长(详细教程)

    怎样申请抖音团长(详细教程)

    微客云 http://www.wikeyun.cn/在说怎样申请抖音团长之前,我简单聊聊我现在对抖音团长的基础认知,先截一张图,这是抖音团长后台的近30天排名截图,是对外公开的。这个图可
    2021-11-06
  • 一个新手如何做直播?主播有哪些聊天技巧?

    一个新手如何做直播?主播有哪些聊天技巧?

    除了多笑,新手主播也要考虑更多丰富的表情和动作:比如适当的剪刀手卖萌,手比爱心的温馨,吐舌头的调皮。介绍产品中途也可以唱唱歌,增加一些灵动的小手势和表情,这会让
    2021-11-06
  • 一个新手如何做直播?主播有哪些聊天技巧?

    一个新手如何做直播?主播有哪些聊天技巧?

    除了多笑,新手主播也要考虑更多丰富的表情和动作:比如适当的剪刀手卖萌,手比爱心的温馨,吐舌头的调皮。介绍产品中途也可以唱唱歌,增加一些灵动的小手势和表情,这会让
    2021-11-06
  • 微客云话费充值系统,可提供充值API接口

    微客云话费充值系统,可提供充值API接口

    微客云提供话费充值系统,电费充值系统,油卡充值系统等等生活缴费系统微客云地址:http://www.wikeyun.cn/话费充值接口文档接口版本:1.0 ―、引言1.1 文档概述本文档提供
    2021-11-06
  • 话费充值API接口,电费充值接口,油卡充值API接口

    话费充值API接口,电费充值接口,油卡充值API接口

    话费充值接口文档接口版本:1.0 ―、引言1.1 文档概述本文档提供话费充值接口规范说明,提供一整套的完整的接入示例(http 接口)供商户参 考,可以帮助商户开发人员快速完成
    2021-11-06

最新评论