Pytorch之torch.meshgrid()

说明：

　　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]])

 转载：https://www.cnblogs.com/regain/p/14494885.html

标签：元素,tensor,torch,张量,Pytorch,meshgrid,print,输入
来源： https://blog.csdn.net/qq_41955460/article/details/123640068