我有这两个张量
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)
解决了
相关问题 更多 >
编程相关推荐