torch.meshgrid()函数

import torch
yv, xv = torch.meshgrid(torch.arange(0, 7), torch.arange(0, 7))
print("xv:","\n__\n", xv ,"\nyv:\n__\n", yv)

xv, yv = xv.flatten().float(),yv.flatten().float()
print("xv:","\n__\n", xv ,"\nyv:\n__\n", yv)

xv: 
__
 tensor([[0, 1, 2, 3, 4, 5, 6],
        [0, 1, 2, 3, 4, 5, 6],
        [0, 1, 2, 3, 4, 5, 6],
        [0, 1, 2, 3, 4, 5, 6],
        [0, 1, 2, 3, 4, 5, 6],
        [0, 1, 2, 3, 4, 5, 6],
        [0, 1, 2, 3, 4, 5, 6]]) 
yv:
__
 tensor([[0, 0, 0, 0, 0, 0, 0],
        [1, 1, 1, 1, 1, 1, 1],
        [2, 2, 2, 2, 2, 2, 2],
        [3, 3, 3, 3, 3, 3, 3],
        [4, 4, 4, 4, 4, 4, 4],
        [5, 5, 5, 5, 5, 5, 5],
        [6, 6, 6, 6, 6, 6, 6]])
xv: 
__
 tensor([0., 1., 2., 3., 4., 5., 6., 0., 1., 2., 3., 4., 5., 6., 0., 1., 2., 3.,
        4., 5., 6., 0., 1., 2., 3., 4., 5., 6., 0., 1., 2., 3., 4., 5., 6., 0.,
        1., 2., 3., 4., 5., 6., 0., 1., 2., 3., 4., 5., 6.]) 
yv:
__
 tensor([0., 0., 0., 0., 0., 0., 0., 1., 1., 1., 1., 1., 1., 1., 2., 2., 2., 2.,
        2., 2., 2., 3., 3., 3., 3., 3., 3., 3., 4., 4., 4., 4., 4., 4., 4., 5.,
        5., 5., 5., 5., 5., 5., 6., 6., 6., 6., 6., 6., 6.])