布尔索引

可以快速上手的开发文档

#布尔索引

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

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