0
点赞
收藏
分享

微信扫一扫

torch.meshgrid

乱世小白 2022-02-12 阅读 159

原理

```python torch.meshgrid(*tensors) ``` 这个是干嘛的?根据输入来创建网格的,例如你告诉我两个列表[1,2],[3],我们可以在二维直角坐标系上标上两个点(1,3),(2,3)。从我们学过的数学的角度来说,前面那个就是x定义域,后面那个就是y定义域。

另外一个更加形象的如下:
在这里插入图片描述
可以看到就是创建了一个网格,所以叫做meshgrid,其中mesh:网状物,grid:格子。

补充一点的是,在我们上面的例子中,其输出是(1,3),(2,3)。但是在pytorch中还要进行进一步转化,变成[1,2],[3,3]。相当于第一个x坐标,第二个放置y坐标。


官方文档:
输入:
在这里插入图片描述

输出:

在这里插入图片描述

看个例子最容易明白:

在这里插入图片描述

举报

相关推荐

0 条评论