9.高阶操作

torch.where（condition,x,y）

是根据条件选取A还是B
三个参数，（condition，A，B），生成的C可能来自于A（直接复制）或B，部分来源于A可以C【0】= A【0】，C【1】=A【1】，但若是从C【0，0，1】=A【0，0，1】，C【0，0，2】=B【0，0，2】等，不像之前那么有规律时，可以写到condition表中。在从condition表中，1的地方来自于A。

condition里面的值越高，取自A的概率越高，给一个条件大于
0.5取自A。
for i
for y
if c[i][j]=A[i][j]
这是由cpu控制的，非常不并行。我们只生成这样一个condition矩阵，我们的赋值语句可以高度并行化，最好的是都用采用GPU。

torch.gather（input,dim,index）

查表的操作
为什么需要呢，价格参照表A，你需要买的东西索引表B，生成的C需要根据B照着A生成物品，根据B中的索引0在A中收集索引为0的物品名字。
gather三个参数，第一个是我们要查的原始表，第二个是维度，第三个是我们需要的索引表，然后输出一个按照索引表排序的原始表内容。
out[i][j][k]=input[i][j][index[i][j][k] #if dim=2