PyTorch入门实战教程笔记（九）：基础张量操作5

PyTorch入门实战教程笔记（九）：基础张量操作5：高阶操作 where&gather 1. Where torch.where(condition, x, y) -->tensor，一共3个参数，第一个参数条件，判断其Ture or False。 如下： 比如，现在将A, B的值根据condition传递给C，可以从C[0]=A[0],C[1]=B[1], 此外，设计到具体的数值，也可以进行替

>>阅读原文<<