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], 此外，設計到具體的數值，也可以進行替

>>阅读原文<<