前言

1. torch.where

 返回满足条件的x/y；

The tensors ​​condition​​​, ​​x​​​, ​​y​​​ must be ​​broadcastable​​.

注意，不同pytorch版本的输入输出数据类型有可能不同。1.6版本x/y输入类型只有Tensor，1.9版本是Tensor or Scalar。

2. 满足条件的索引

3. 满足条件的元素赋值

参考

1. ​​torch.where​​;

2. ​​Pytorch tensor特定条件判断索引​​;

完