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

转载 作者：qq735679552 更新时间：2022-09-28 22:32:09 46 4

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]])

46 4 0

• 详解C语言sscanf()函数、vsscanf()函数、vscanf()函数

  C语言sscanf()函数：从字符串中读取指定格式的数据 头文件： ?

• php - 如何解释at()函数； substr()函数;伪代码中的exist()函数

  最近，我有一个关于工作预评估的问题，即使查询了每个功能的工作原理，我也不知道如何解决。这是一个伪代码。 下面是一个名为foo()的函数，该函数将被传递一个值并返回一个值。如果将以下值传递给foo函数，

• VBS教程：函数-CStr 函数

  CStr 函数 返回表达式，该表达式已被转换为 String 子类型的 Variant。 CStr(expression) expression 参数是任意有效的表达式。 说明 通常，可以

• VBS教程：函数-CSng 函数

  CSng 函数 返回表达式，该表达式已被转换为 Single 子类型的 Variant。 CSng(expression) expression 参数是任意有效的表达式。 说明 通常，可

• VBS教程：函数-CreateObject 函数

  CreateObject 函数 创建并返回对 Automation 对象的引用。 CreateObject(servername.typename [, location]) 参数 serv

• VBS教程：函数-Cos 函数

  Cos 函数 返回某个角的余弦值。 Cos(number) number 参数可以是任何将某个角表示为弧度的有效数值表达式。 说明 Cos 函数取某个角并返回直角三角形两边的比值。此比值是

• VBS教程：函数-CLng 函数

  CLng 函数 返回表达式，此表达式已被转换为 Long 子类型的 Variant。 CLng(expression) expression 参数是任意有效的表达式。 说明 通常，您可以使

• VBS教程：函数-CInt 函数

  CInt 函数 返回表达式，此表达式已被转换为 Integer 子类型的 Variant。 CInt(expression) expression 参数是任意有效的表达式。 说明 通常，可

• VBS教程：函数-Chr 函数

  Chr 函数 返回与指定的 ANSI 字符代码相对应的字符。 Chr(charcode) charcode 参数是可以标识字符的数字。 说明 从 0 到 31 的数字表示标准的不可打印的

• VBS教程：函数-CDbl 函数

  CDbl 函数 返回表达式，此表达式已被转换为 Double 子类型的 Variant。 CDbl(expression) expression 参数是任意有效的表达式。 说明 通常，您可

• VBS教程：函数-CDate 函数

  CDate 函数 返回表达式，此表达式已被转换为 Date 子类型的 Variant。 CDate(date) date 参数是任意有效的日期表达式。 说明 IsDate 函数用于判断 d

• VBS教程：函数-CCur 函数

  CCur 函数 返回表达式，此表达式已被转换为 Currency 子类型的 Variant。 CCur(expression) expression 参数是任意有效的表达式。 说明 通常，

• VBS教程：函数-CByte 函数

  CByte 函数 返回表达式，此表达式已被转换为 Byte 子类型的 Variant。 CByte(expression) expression 参数是任意有效的表达式。 说明 通常，可以

• VBS教程：函数-CBool 函数

  CBool 函数 返回表达式，此表达式已转换为 Boolean 子类型的 Variant。 CBool(expression) expression 是任意有效的表达式。 说明 如果 ex

• VBS教程：函数-Atn 函数

  Atn 函数 返回数值的反正切值。 Atn(number) number 参数可以是任意有效的数值表达式。 说明 Atn 函数计算直角三角形两个边的比值 (number) 并返回对应角的弧

• VBS教程：函数-Asc 函数

  Asc 函数 返回与字符串的第一个字母对应的 ANSI 字符代码。 Asc(string) string 参数是任意有效的字符串表达式。如果 string 参数未包含字符，则将发生运行时错误。

• VBS教程：函数-Array 函数

  Array 函数 返回包含数组的 Variant。 Array(arglist) arglist 参数是赋给包含在 Variant 中的数组元素的值的列表（用逗号分隔）。如果没有指定此参数，则

• VBS教程：函数-Abs 函数

  Abs 函数 返回数字的绝对值。 Abs(number) number 参数可以是任意有效的数值表达式。如果 number 包含 Null，则返回 Null；如果是未初始化变量，则返回 0。

• VBS教程：函数-FormatPercent 函数

  FormatPercent 函数 返回表达式，此表达式已被格式化为尾随有 % 符号的百分比（乘以 100 ）。 FormatPercent(expression[,NumDigitsAfterD

• VBS教程：函数-FormatNumber 函数

  FormatNumber 函数 返回表达式，此表达式已被格式化为数值。 FormatNumber(   expression [,NumDigitsAfterDecimal [,Inc