pytorch中gather的用法

最新推荐文章于 2023-06-05 00:16:11 发布

SnowMaple_Queen

最新推荐文章于 2023-06-05 00:16:11 发布

阅读量414

点赞数

分类专栏： pytorch 文章标签：索引

本文链接：https://blog.csdn.net/qq_38558834/article/details/109287349

版权

这篇博客介绍了PyTorch中的gather函数，包括其官方解释和参数解析。gather函数用于根据指定的索引从张量中提取特定值，其中输入参数包括原张量、索引维度和目标索引位置。例如，在按列索引的情况下，[[0, 0], [1, 0]]表示取第一列的第一个元素和第二列的第一个元素。" 125643714,13029872,Java高级框架面试题解析,"['Java', '面试', '框架', '开发工具', '分布式']

摘要由CSDN通过智能技术生成

官方解释

def gather(self, input, dim, index, out=None): # real signature unknown; restored from __doc__      """
        gather(input, dim, index, out=None) -> Tensor
        Gathers values along an axis specified by `dim`.   
        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     
        If :attr:`input` is an n-dimensional tensor with size
        :math:`(x_0, x_1..., x_{i-1}, x_i, x_{i+1}, ..., x_{n-1})`
        and ``dim = i``, then :attr:`index` must be an :math:`n`-dimensional tensor with
        size :math:`(x_0, x_1, ..., x_{i-1}, y, x_{i+1}, ..., x_{n-1})` where :math:`y \geq 1`
        and :