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

最新推荐文章于 2025-03-24 11:04:26 发布

devil_son1234

最新推荐文章于 2025-03-24 11:04:26 发布

阅读量2k

点赞数 2

文章标签： pytorch 深度学习 python

原文链接：https://blog.youkuaiyun.com/qq_41955460/article/details/123640068

版权

torch.meshgrid()用于生成网格坐标，输入为两个一维tensor，输出为两个tensor，行数和列数分别对应输入tensor的元素数量。当输入数据类型不匹配或维度非一维时会抛出错误。输出的第一个tensor的每一行填充第一个输入的元素，第二个tensor的每一列填充第二个输入的元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

说明：
　　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]])