Pytorch之torch.meshgrid()
说明:
torch.meshgrid()的功能是生成网格,可以用于生成坐标。
函数输入:
输入两个数据类型相同的一维tensor
函数输出:
输出两个tensor(tensor行数为第一个输入张量的元素个数,列数为第二个输入张量的元素个数)
注意:
1)当两个输入tensor数据类型不同或维度不是一维时会报错。
2)其中第一个输出张量填充第一个输入张量中的元素,各行元素相同;第二个输出张量填充第二个输入张量中的元素各列元素相同。<填充效果见实验结果>
实验验证:
1 import torch 2 x = torch.tensor([1,2,3]) #定义以为张量x,作为行输入张量 3 y = torch.tensor([5,6,7]) #定义以为张量y,作为行列入张量 4 resultx,resulty = torch.meshgrid(x,y) 5 print(x,y) 6 print(resultx) 7 print(resulty)
实验结果(jupyter notebook):