torch stack() meshgrid()

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

# Input is a one-dimensional sequence , Output two 2D meshes , Commonly used to generate coordinates 
ox,oy = torch.meshgrid([x,y])

# Splice according to a certain dimension , Input sequence shape It has to be consistent , The default in accordance with the 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]])
amount to (0,0)(0,1)… Two pairs of elements with the same position of two-dimensional tensors are one coordinate
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])