【Pytorch小知识】torch.gather()函数的用法及在Focal Loss中的应用(详细易懂)
文章目录官方文档中torch.gather的用法torch.gather应用总结官方文档中torch.gather的用法torch.gather(input,dim,index,out=None)→TensorGathersvaluesalonganaxisspecifiedbydim.Fora3-Dtensortheoutputisspecifiedby:out[i][j][k]=input[i