代码参考自:LIIF:Learning Continuous Image Representation with Local Implicit Image Function
其中shape表示想要生成的网格的大小,比如可以输入shape=(512,512)则最后生成的是512x512大小的表格。
着重强调的地方在于x,y = torch.meshgrid(a,b)返回的是两个张量,两个张量的形状是一样的,即a的张量的长度决定着新的张量的行数,而b的张量的长度决定着新的张量的列数,所以新的张量的形状是(a.shape,b.shape),这样,再利用torch.stack吧x,y在最后一列拼起来就是一个表格的坐标系了。