tensor.fip()操作

anchors = torch.tensor([[10, 13, 16, 30, 33, 23],
           [30, 61, 62, 45, 59, 119],  
           [116, 90, 156, 198, 373, 326]])
anchors[:] = anchors.flip(0)
print(anchors)

 在第1维上翻转:

anchors[:] = anchors.flip(1)
print(anchors)

猜你喜欢

转载自blog.csdn.net/weixin_48747603/article/details/127704992
今日推荐