会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
jayson
博客园
首页
新随笔
联系
管理
2020年3月29日
pytorch高阶OP操作where,gather
摘要: 一、where 1)torch.where(condition, x, y) # condition是条件,满足条件就返回x,不满足就返回y 2)特点,相比for循环的优点是:可以布置在GPU上运行 二、gather 1)官方解释:根据指定的维度和索引值来筛选值 2)举例
阅读全文
posted @ 2020-03-29 16:57 jaysonteng
阅读(2713)
评论(1)
推荐(0)
编辑