PyTorch——(4)where条件判断、gather查表

where() 条件判断

gather()查表

input ：待查的表
dim ： 在input的哪个维度上查表
index： 待查表的索引值