没有合适的资源?快使用搜索试试~ 我知道了~
Pytorch的gather()和scatter()
0 下载量 98 浏览量
2021-01-21
16:37:31
上传
评论
收藏 62KB PDF 举报
温馨提示
试读
1页
Pytorch的gather()和scatter() 1.gather() gather是取的意思,意为把某一tensor矩阵按照一个索引序列index取出,组成一个新的矩阵。 gather(input,dim,index) 参数: input是要取值的矩阵 dim指操作的维度,0为竖向操作即按行操作,1为横向操作即按列操作 index为索引序列 下面这个例子是按行取出第一行的’0号元素’,’0行元素’组成新的第一行; 再取出第二行的‘1号元素’,‘0号元素’组成新的第二行 a = torch.Tensor([[1,2],[3,4]]) b = torch.gather(a, 1, torch
资源推荐
资源评论
资源评论
weixin_38501299
- 粉丝: 3
- 资源: 923
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功