机器学习PyTorch——3.高阶OP

一.where函数

torch.where(condition, x, y)

where函数具体作用为：利用x和y生成一个新的tensor，其中的参数 condition 为一个与x和y一样shape的tensor，掌管着新的tensor的生成条件，如果condition某个位置的值为1，则新的tensor来自与x， 反之为0，则来自于y。（condition的使用很灵活）例如：

二.gather函数

torch.gather(input, dim, index, out=None)

该函数作用为查表，其中input作为参照表，使用dim来指定维度，index则是你要查找一一的应的tensor，最后将index以input的方式进行返回。具体使用方式及规则如下：

例如：