torch张量法的UNSQUEZE和扩展的numpy等价码

2024-03-28 12:09:01 发布

您现在位置:Python中文网/ 问答频道 /正文

我有这两个张量

box_a = torch.randn(1,4)
box_b = torch.randn(1,4)

我在pytorch中有一个代码

box_a[:, 2:].unsqueeze(1).expand(1, 1, 2)

但是我想用numpy转换上面的代码
对于box_abox_b我可以这样做

  box_a = numpy.random.randn(1,4)
  box_b = numpy.random.randn(1,4)

但是这个呢

box_a[:, 2:].unsqueeze(1).expand(1, 1, 2)

Tags: 代码numpyboxrandomtorchpytorchexpandrandn