PyTorch | 高阶操作

1. where 函数

• torch.where(condition, x, y)：$condition$ 是条件，$x$ 和 $y$ 是同 $shape$ 的矩阵，针对 $x$ 矩阵中的每个元素，如果满足 $condition$ 条件就保留元素，不满足 $condition$ 条件就使用 $y$ 矩阵对应位置的元素值进行替换，最后返回结果。
  

2. gather 函数

• torch.gather(input, dim, index, out=None)：https://blog.csdn.net/m0_52650517/article/details/119389655#t7