【Torch API】pytorch 中meshgrid()函数详解

说明：
　　torch.meshgrid()的功能是生成网格，可以用于生成坐标。

函数输入:
　　输入两个数据类型相同的一维tensor

函数输出：
输出两个tensor（tensor行数为第一个输入张量的元素个数，列数为第二个输入张量的元素个数）

注意：
　　1）当两个输入tensor数据类型不同或维度不是一维时会报错。

　　2）其中第一个输出张量填充第一个输入张量中的元素，各行元素相同；第二个输出张量填充第二个输入张量中的元素各列元素相同。
 

CH19

于 2022-03-21 16:58:13 发布

2312
 收藏
分类专栏： 解决方案 文章标签： python
版权

解决方案
专栏收录该内容
3 篇文章1 订阅
订阅专栏
说明：
　　torch.meshgrid()的功能是生成网格，可以用于生成坐标。

函数输入:
　　输入两个数据类型相同的一维tensor

函数输出：
输出两个tensor（tensor行数为第一个输入张量的元素个数，列数为第二个输入张量的元素个数）

注意：
　　1）当两个输入tensor数据类型不同或维度不是一维时会报错。

　　2）其中第一个输出张量填充第一个输入张量中的元素，各行元素相同；第二个输出张量填充第二个输入张量中的元素各列元素相同。

实验验证：

import torch
x = torch.tensor([1,2,3]) #定义以为张量x，作为行输入张量
y = torch.tensor([5,6,7]) #定义以为张量y，作为行列入张量
resultx,resulty = torch.meshgrid(x,y)
print(x,y)
print(resultx)
print(resulty)

实验结果（jupyter notebook）：

tensor([1, 2, 3]) tensor([5, 6, 7])
tensor([[1, 1, 1],
        [2, 2, 2],
        [3, 3, 3]])
tensor([[5, 6, 7],
        [5, 6, 7],
        [5, 6, 7]])