【pytorch】从yolo的make_grid理解torch.meshgrid、torch.stack
文章目录简述1、torch.meshgrid创建行列坐标2、torch.stack结合行列坐标3、通过view函数扩展维度简述yolo检测make_grid创建网格代码如下,那么什么是torch.meshgrid?def_make_grid(nx=20,ny=20):yv,xv=torch.meshgrid([torch.arange(ny),torch.arange(nx)])returntor