wh1:
wh2:
wh1 = wh1[:, None]
wh2 = wh2[None]
wh1:
wh2:
二. 比较大小 2.1 torch.min(只需要最后一维shape相同即可)3x1x2维的张量 与 1x3x2维的张量比较(使用torch.min(wh1,wh2))(下图右侧)
<

wh1:
wh2:
wh1 = wh1[:, None]
wh2 = wh2[None]
wh1:
wh2:
二. 比较大小 2.1 torch.min(只需要最后一维shape相同即可)3x1x2维的张量 与 1x3x2维的张量比较(使用torch.min(wh1,wh2))(下图右侧)
<