关于pytorch中的[...,n]

今天在看yolov4代码时看到的疑问，自己便实验了一下

以prediction维度为batchsize*3*13*13*85为例，prediction[...,0]即取第一个轴上的所有数据，即维度为batchsize*3*13*13。

yolov4中85分别代表先验框的x,y坐标，宽，高，是否包括物体，以及80个类别概率

prediction[...,0]对应x的信息

prediction[...,1]对应y的信息

prediction[...,2]对应宽的信息

prediction[...,3]对应高的信息