pytorch（7）Where和Gather
- Where
torch.where(condition,x,y) →Tensor x是源头A，y是源头B有两个数据A和B，生成了C。
C有可能来自于A，有可能部分来自A，C于A、B的对应关系规则写在condition里。
out=x if condition
out=y if othewise
实际含义：概率大于0.5时取A的对应位置的值
- gather 生成一张表
torch.gather(input,dim,index,out=None) 收集，查表过程
[dog cat fish]→ input dim=0 [1 0 1 2]→ index
输出[cat dog cat fish]