会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
Mr.Higgerw
越努力,越幸运
博客园
首页
联系
管理
2020年11月28日
pytorch index_select()函数
摘要: 函数实现从当前张量中从某个维度选择一部分序号的张量 tensor.select_index(dim, index) 对于一个二维张量feature: 第一个参数 参数0表示按行索引,1表示按列进行索引 第二个参数 是一个整数类型的一维tensor,就是索引的序号 二维张量举例: 三维张量举例: 另一
阅读全文
posted @ 2020-11-28 11:34 爱吃砂糖橘的白龙
阅读(1109)
评论(0)
推荐(0)
编辑
公告