当前位置：代码迷 >> 综合 >> torch.where()，torch.gather()

详细解决方案

torch.where()，torch.gather()

热度：13 发布时间：2024-02-06 09:28:49.0

torch.where()

给定一个条件cond，满足条件的取x对应位置元素，不满足的取y对应元素

torch.gather()

实际上是一个查表操作

先计算出满足条件的索引index，再利用得到的index从需要查找的数据中进行查找

相关解决方案