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

where() 条件判断

![在这里插入图片描述]( https://img-blog.csdnimg.cn/20210425205924265.png?x-oss-process=image/watermark ,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM0MjEzMjYw,size_16,color_FFFFFF,t_70)![在这里插入图片描述]( https://img-blog.csdnimg.cn/20210425205941338.png?x-oss-process=image/watermark ,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM0MjEzMjYw,size_16,color_FFFFFF,t_70)

gather()查表

input ：待查的表
dim ：在input的哪个维度上查表
index：待查表的索引值
![在这里插入图片描述]( https://img-blog.csdnimg.cn/20210425205859943.png?x-oss-process=image/watermark ,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM0MjEzMjYw,size_16,color_FFFFFF,t_70)

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

刷新页面返回顶部

Live2D