深度学习—高阶操作

where函数

深度学习—高阶操作_第1张图片

 x y 共同组成z 可能在z[0]=x[0],z[1]=y[1],满足condition就用x,不满足就选y

深度学习—高阶操作_第2张图片

 

Gather函数

深度学习—高阶操作_第3张图片

gather 相当于查表  

深度学习—高阶操作_第4张图片

index=idx.long() => [4,3]的tensor ,里面的值为[[7,4,9],[7,4,9],[...],...]

用idx查expan里第7个第4个第9个...

得到[[107,104,109],[107,104,109],...]

 

你可能感兴趣的:(深度学习,深度学习)