引言: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