pytorch笔记 meshgrid

Posted 2022-09-26 UQI-LIUWJ

1 基本使用方法

• 生成网格
• 输入两个数据类型相同的一维tensor
• 输出两个tensor（tensor行数为第一个输入张量的元素个数，列数为第二个输入张量的元素个数）
  • 第一个输出张量填充第一个输入张量中的元素，各行元素相同
  • 第二个输出张量填充第二个输入张量中的元素，各列元素相同。

2 举例说明

import torch
coords_h = torch.arange(2)
coords_w = torch.arange(3)
torch.meshgrid([coords_h, coords_w])
'''
(tensor([[0, 0, 0],
         [1, 1, 1]]),
 tensor([[0, 1, 2],
         [0, 1, 2]]))
'''