从 torch 张量方法解压缩和扩展的 numpy 等效代码

Question

我有这两个张量

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

我在 pytorch 中有一个代码

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

但我想将上面的代码转换为 numpy
对于 box_a 和 box_b 我可以做这样的事情

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

但是这个呢

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

Answer 1

解决了

box_a = np.random.randn(1,4)
box_b = np.random.randn(1,4)
max_xy = np.broadcast_to(np.expand_dims(box_a[:, 2:],axis=1),(1,1,2))

numpy equivalent code of unsqueeze and expand from torch tensor method