torch stack() meshgrid()

x= torch.arange(4)
y= torch.arange(4)

#输入为一维序列，输出两个二维网格，常用来生成坐标
ox,oy = torch.meshgrid([x,y])

#按照某个维度拼接，输入序列shape必须一致，默认按照dim0
o2 = torch.stack((ox,oy))

print(ox,oy)
print(o2,o2.shape)

tensor([[0, 0, 0, 0],
[1, 1, 1, 1],
[2, 2, 2, 2],
[3, 3, 3, 3]])
tensor([[0, 1, 2, 3],
[0, 1, 2, 3],
[0, 1, 2, 3],
[0, 1, 2, 3]])
相当于(0,0)(0,1)…两个二维张量相同位置元素对就是一个坐标
tensor([[[0, 0, 0, 0],
[1, 1, 1, 1],
[2, 2, 2, 2],
[3, 3, 3, 3]],

    [[0, 1, 2, 3],
     [0, 1, 2, 3],
     [0, 1, 2, 3],
     [0, 1, 2, 3]]]) torch.Size([2, 4, 4])