PyTorch——(4)where条件判断、gather查表 - 一抹烟霞 - 博客园

PyTorch——(4)where条件判断、gather查表

where() 条件判断

在这里插入图片描述

gather()查表

在这里插入图片描述
input ：待查的表
dim ：在input的哪个维度上查表
index：待查表的索引值

posted on 2021-04-25 21:01 一抹烟霞阅读(592) 评论(0) 编辑收藏举报

刷新页面返回顶部

Live2D