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