1
|
官方文档: |
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
torch.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] # dim=0 out[i][j][k] = input [i][index[i][j][k]][k] # dim=1 out[i][j][k] = input [i][j][index[i][j][k]] # dim=2 Parameters: input (Tensor) - The source tensor dim ( int ) - The axis along which to index index (LongTensor) - The indices of elements to gather out (Tensor, optional) - Destination tensor Example: >>> t = torch.Tensor([[ 1 , 2 ],[ 3 , 4 ]]) >>> torch.gather(t, 1 , torch.LongTensor([[ 0 , 0 ],[ 1 , 0 ]])) 1 1 4 3 [torch.FloatTensor of size 2x2 ] |
栏目列表
最新更新
nodejs爬虫
Python正则表达式完全指南
爬取豆瓣Top250图书数据
shp 地图文件批量添加字段
爬虫小试牛刀(爬取学校通知公告)
【python基础】函数-初识函数
【python基础】函数-返回值
HTTP请求:requests模块基础使用必知必会
Python初学者友好丨详解参数传递类型
如何有效管理爬虫流量?
2个场景实例讲解GaussDB(DWS)基表统计信息估
常用的 SQL Server 关键字及其含义
动手分析SQL Server中的事务中使用的锁
openGauss内核分析:SQL by pass & 经典执行
一招教你如何高效批量导入与更新数据
天天写SQL,这些神奇的特性你知道吗?
openGauss内核分析:执行计划生成
[IM002]Navicat ODBC驱动器管理器 未发现数据
初入Sql Server 之 存储过程的简单使用
SQL Server -- 解决存储过程传入参数作为s
关于JS定时器的整理
JS中使用Promise.all控制所有的异步请求都完
js中字符串的方法
import-local执行流程与node模块路径解析流程
检测数据类型的四种方法
js中数组的方法,32种方法
前端操作方法
数据类型
window.localStorage.setItem 和 localStorage.setIte
如何完美解决前端数字计算精度丢失与数