pytorch中torch.meshgrid()函数理解及举例说明

说明:

函数的功能是生成网格,可以用于生成坐标。

函数输入:

输入两个一维tensor数据,且两个tensor数据类型相同,也可以输入三个一维tensor数据

函数输出:

输出两个tensor数据(两个tensor的行数为第一个输入张量的元素个数,列数为第二个输入张量的元素个数)或者三个tensor数据(三个tensor第一维度大小为第一个输入张量的元素个数,第二维度大小为第二个输入张量的元素个数,第三维度为第三个输入张量元素个数)

报错:

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

结果理解:

输入两个一维张量的元素个数分别为n1,n2,则输出两个张量是二维的,且行和列个数均为n1,n2,输出第一个张量行相同(对应第一个输入张量),输出第二个张量列相同(对应第二个输入张量),其中第一个输出张量填充第一个输入张量中的元素,各行元素相同 ;第二个输出张量填充第二个输入张量中的元素,各列元素相同

若输入是三个一维张量,元素个数分别为n1,n2,n3,则输出的三个张量都是三维的,且输出的三个张量的三个维度均相等,分别为n1,n2,n3。

输入为两个张量:

python 复制代码
import torch
import torch.nn as nn

a1 = torch.tensor([1,3])
b1 = torch.tensor([2,4,6])
x1,y1 = torch.meshgrid(a1,b1)
print(x1)
print(y1)

输出:
tensor([[1, 1, 1],
        [3, 3, 3]])
tensor([[2, 4, 6],
        [2, 4, 6]])

输入为三个张量:

python 复制代码
import torch
import torch.nn as nn

a2 = torch.tensor([1,3])
b2 = torch.tensor([2,4,6])
c2 = torch.tensor([7,8,9,10])
x2,y2,z2 = torch.meshgrid(a2,b2,c2)
print(x2)
print(x2.shape)
print(y2)
print(y2.shape)
print(z2)
print(z2.shape)

输出:
tensor([[[1, 1, 1, 1],
         [1, 1, 1, 1],
         [1, 1, 1, 1]],

        [[3, 3, 3, 3],
         [3, 3, 3, 3],
         [3, 3, 3, 3]]])
torch.Size([2, 3, 4])
tensor([[[2, 2, 2, 2],
         [4, 4, 4, 4],
         [6, 6, 6, 6]],

        [[2, 2, 2, 2],
         [4, 4, 4, 4],
         [6, 6, 6, 6]]])
torch.Size([2, 3, 4])
tensor([[[ 7,  8,  9, 10],
         [ 7,  8,  9, 10],
         [ 7,  8,  9, 10]],

        [[ 7,  8,  9, 10],
         [ 7,  8,  9, 10],
         [ 7,  8,  9, 10]]])
torch.Size([2, 3, 4])
相关推荐
学术小八34 分钟前
2025年人工智能、虚拟现实与交互设计国际学术会议
人工智能·交互·vr
仗剑_走天涯2 小时前
基于pytorch.nn模块实现线性模型
人工智能·pytorch·python·深度学习
cnbestec3 小时前
协作机器人UR7e与UR12e:轻量化设计与高负载能力助力“小而美”智造升级
人工智能·机器人·协作机器人·ur协作机器人·ur7e·ur12e
zskj_zhyl3 小时前
毫米波雷达守护银发安全:七彩喜跌倒检测仪重构居家养老防线
人工智能·安全·重构
gaosushexiangji4 小时前
利用sCMOS科学相机测量激光散射强度
大数据·人工智能·数码相机·计算机视觉
ai小鬼头5 小时前
AIStarter新版重磅来袭!永久订阅限时福利抢先看
人工智能·开源·github
说私域5 小时前
从品牌附庸到自我表达:定制开发开源AI智能名片S2B2C商城小程序赋能下的营销变革
人工智能·小程序
飞哥数智坊6 小时前
新版定价不够用,Cursor如何退回旧版定价
人工智能·cursor
12点一刻6 小时前
搭建自动化工作流:探寻解放双手的有效方案(2)
运维·人工智能·自动化·deepseek
未来之窗软件服务6 小时前
东方仙盟AI数据中间件使用教程:开启数据交互与自动化应用新时代——仙盟创梦IDE
运维·人工智能·自动化·仙盟创梦ide·东方仙盟·阿雪技术观