栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

torch.meshgrid的作用

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

torch.meshgrid的作用

假设有如下的网格:

由上图可知,x轴的范围是从0-1,y轴的范围是从0-3
使用以下代码生成x,y的范围:

x_range = torch.arange(0,2)
y_range = torch.arange(0,4)

而torch.meshgrid可以让我们在以某个维度顺序依次获得各个点的x坐标值,和y坐标轴。

当x轴坐标范围放在第二个参数中,就一行一行的去获得各个点的x坐标值,和y坐标轴

y,x = torch.meshgrid(y_range,x_range)
print(x)
print(y)

输出则是,结合图可知,上述操作获得了按照x轴依次获得各个点的横纵坐标
举例来说,访问的第一个点为(0,0),则x中的第一行的第一个值为0;接着访问的下一个点为(1,0),则x中的第一行的第二个值为1,同理y中的第一行全为0

tensor([[0, 1],
        [0, 1],
        [0, 1],
        [0, 1]])
tensor([[0, 0],
        [1, 1],
        [2, 2],
        [3, 3]])

当y轴坐标范围放在第二个参数中,就一列一列的去获得各个点的x坐标值,和y坐标轴
举例来说,访问的第一个点为(0,0),则x中的第一行的第一个值为0;接着访问的下一个点为(0,1),则x中的第一行的第二个值为0,再接下来访问(0,2),直至(0,3),而这个过程中,所有的横坐标都为0。

x,y = torch.meshgrid(x_range,y_range)
print(x)
print(y)

结果如图所示

tensor([[0, 0, 0, 0],
        [1, 1, 1, 1]])
tensor([[0, 1, 2, 3],
        [0, 1, 2, 3]])
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/313798.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号