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]