torch.where

基于torch.where和布尔索引的速度比较

我就废话不多说了，直接上代码吧！ import torch import time x = torch.Tensor([[1, 2, 3], [5, 5, 5], [7, 8, 9],[5,5,5