torch.meshgrid()函数《pytorch学习》

引言：torch.meshgrid（）函数的作用在下只理解出一层含义，就是数据扩张，就是把一维数据扩张，然后可以和其他数据联合，，可以构成网状数据。

1.数据扩张。

        在下理解就是x轴的数据扩张成三个x轴数据，y轴的数据扩张成三个y轴数据，然后可构成一张3*3大小的网状，可以利用索引取相应的值。例如;

代码：

data1=torch.tensor([2,5,6])
print(data1)
data2=torch.tensor([6,9,4])
print(data2)
gx,gy=torch.meshgrid(data1,data2)
print(gx)
print(gy)

output：

tensor([2, 5, 6])
tensor([6, 9, 4])
tensor([[2, 2, 2],
        [5, 5, 5],
        [6, 6, 6]])
tensor([[6, 9, 4],
        [6, 9, 4],
        [6, 9, 4]])

2.数据曲面构建

代码：

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D


xs = torch.linspace(-5, 5, steps=100)
ys = torch.linspace(-5, 5, steps=100)
x, y = torch.meshgrid(xs, ys)
z = torch.sin(torch.sqrt(x