gpt4 book ai didi

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

转载 作者:qq735679552 更新时间:2022-09-28 22:32:09 46 4
gpt4 key购买 nike

CFSDN坚持开源创造价值,我们致力于搭建一个资源共享平台,让每一个IT人在这里找到属于你的精彩世界.

这篇CFSDN的博客文章解析Pytorch中的torch.gather()函数由作者收集整理,如果你对这篇文章有兴趣,记得点赞哟.

参数说明

以官方说明为例,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]])

  。

画图说明

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

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

官方文档 。

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()函数内容请搜索我以前的文章或继续浏览下面的相关文章希望大家以后多多支持我! 。

原文链接:https://blog.csdn.net/xiaoliujun1999/article/details/121292061 。

最后此篇关于解析Pytorch中的torch.gather()函数的文章就讲到这里了,如果你想了解更多关于解析Pytorch中的torch.gather()函数的内容请搜索CFSDN的文章或继续浏览相关文章,希望大家以后支持我的博客! 。

46 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com