【pytorch基礎】滿足條件的元素賦值

前言

 

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特定條件判斷索引;

完