torch.meshgrid函数
其作用是对于给定的x,y坐标,返回(x,y)所组合形成的所有点的横,纵坐标
比如下文中x.shape=10
m_x.shape=(10,10)
测试代码如下:
import torch
x=torch.arange(0,10)
y=torch.arange(50,60)
m_x,m_y=torch.meshgrid(x,y)
print(m_x,m_y)
>>>tensor([[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
[2, 2, 2, 2, 2, 2, 2, 2, 2, 2],
[3, 3, 3, 3, 3, 3, 3, 3, 3, 3],
[4, 4, 4, 4, 4, 4, 4, 4, 4, 4],
[5, 5, 5, 5, 5, 5, 5, 5, 5, 5],
[6, 6, 6, 6, 6, 6, 6, 6, 6, 6],
[7, 7, 7, 7, 7, 7, 7, 7, 7, 7],
[8, 8, 8, 8, 8, 8, 8, 8, 8, 8],
[9, 9, 9, 9, 9, 9, 9, 9, 9, 9]]) tensor([[50, 51, 52, 53, 54, 55, 56, 57, 58, 59],
[50, 51, 52, 53, 54, 55, 56, 57, 58, 59],
[50, 51, 52, 53, 54, 55, 56, 57, 58, 59],
[50, 51, 52, 53, 54, 55, 56, 57, 58, 59],
[50, 51, 52, 53, 54, 55, 56, 57, 58, 59],
[50, 51, 52, 53, 54, 55, 56, 57, 58, 59],
[50, 51, 52, 53, 54, 55, 56, 57, 58, 59],
[50, 51, 52, 53, 54, 55, 56, 57, 58, 59],
[50, 51, 52, 53, 54, 55, 56, 57, 58, 59],
[50, 51, 52, 53, 54, 55, 56, 57, 58, 59]])