- 在yolov3中,要将anchor的宽度
[[116],[156],[373]]
复制成batch_size*3*scaled_w*scaled_h的形状,故对repeat的方式做了一下研究。 - 取batch_size=2,scaled_w=scaled_h=5
import torch
w=torch.Tensor([[116],[156],[373]])
u=w.repeat(50,1).view(2,3,5,5)
v=w.repeat(1,50).view(2
[[116],[156],[373]]
复制成batch_size*3*scaled_w*scaled_h的形状,故对repeat的方式做了一下研究。import torch
w=torch.Tensor([[116],[156],[373]])
u=w.repeat(50,1).view(2,3,5,5)
v=w.repeat(1,50).view(2