Tensor operation in pytoch

One . Expand dimensions

1.1. Use unsqueeze

1.2. YOLO Used in [none]

wh1：

wh2：  

    wh1 = wh1[:, None]
    wh2 = wh2[None]

wh1：

wh2： 

Two . Compare the size

2.1 torch.min（ Only the last dimension is needed shape The same can be ）

3x1x2 D tensor And 1x3x2 Tensor comparison of dimensions （ Use torch.min(wh1,wh2)）（ On the right side of the picture below ）

<