torch.meshgrid（）函数学习

meshgrid是用于生成网络的函数

import torch
x_range = torch.arange(0,2)
y_range = torch.arange(0,4)
y,x = torch.meshgrid(y_range,x_range)
print(x)
print(y)

即生成网格为：

同时以按行读取的方式赋值给x,y

由于此时行变换对应的是y值的变换，所以X的值每行对应均为0,1

tensor([[0, 1],
        [0, 1],
        [0, 1],
        [0, 1]])

同理，y的值在行变化时候产生变化，但在同一行值是固定的，因此对应为：

tensor([[0, 0],
        [1, 1],
        [2, 2],
        [3, 3]])

如果代码为x,y时：

import torch
x_range = torch.arange(0,2)
y_range = torch.arange(0,4)
x,y = torch.meshgrid(x_range,y_range)
print(x)
print(y)

生成网格为：

 此时按行读取即按x读取：

所以输出x和y依次分别为：

tensor([[0, 0, 0, 0],
        [1, 1, 1, 1]])
tensor([[0, 1, 2, 3],
        [0, 1, 2, 3]])