快捷方式

torch.meshgrid

torch.meshgrid(*tensors, indexing=None)[source]

根据 attr:tensors 中的 1D 输入创建坐标网格。

当您想要在一个输入范围内可视化数据时,这很有帮助。请参见下面的绘图示例。

给定 NN 个 1D 张量 T0TN1T_0 \ldots T_{N-1} 作为输入,它们对应的尺寸为 S0SN1S_0 \ldots S_{N-1},这将创建 NN 个 N 维张量 G0GN1G_0 \ldots G_{N-1},每个张量的形状为 (S0,...,SN1)(S_0, ..., S_{N-1}),其中输出 GiG_i 是通过将 TiT_i 扩展到结果形状而构建的。

注意

0D 输入等价于单个元素的 1D 输入。

警告

torch.meshgrid(*tensors) 当前的行为与调用 numpy.meshgrid(*arrays, indexing=’ij’) 相同。

将来,torch.meshgrid 将过渡到使用 indexing=’xy’ 作为默认值。

https://github.com/pytorch/pytorch/issues/50276 跟踪此问题,目标是迁移到 NumPy 的行为。

另请参阅

torch.cartesian_prod() 具有相同的效果,但它将数据收集到一个向量张量中。

参数
  • tensors (列表 of 张量) – 标量或 1 维张量的列表。标量将自动视为大小为 (1,)(1,) 的张量

  • indexing (可选[字符串]) –

    (字符串,可选):索引模式,“xy” 或 “ij”,默认为 “ij”。有关未来的更改,请参阅警告。

    如果选择 “xy”,则第一维对应于第二个输入的基数,第二维对应于第一个输入的基数。

    如果选择 “ij”,则维度与输入的基数顺序相同。

返回值

如果输入包含大小为S0SN1S_0 \ldots S_{N-1}NN个张量,那么输出也将包含NN个张量,其中每个张量形状为(S0,...,SN1)(S_0, ..., S_{N-1})

返回类型

seq(张量序列)

示例

>>> x = torch.tensor([1, 2, 3])
>>> y = torch.tensor([4, 5, 6])

Observe the element-wise pairings across the grid, (1, 4),
(1, 5), ..., (3, 6). This is the same thing as the
cartesian product.
>>> grid_x, grid_y = torch.meshgrid(x, y, indexing='ij')
>>> grid_x
tensor([[1, 1, 1],
        [2, 2, 2],
        [3, 3, 3]])
>>> grid_y
tensor([[4, 5, 6],
        [4, 5, 6],
        [4, 5, 6]])

This correspondence can be seen when these grids are
stacked properly.
>>> torch.equal(torch.cat(tuple(torch.dstack([grid_x, grid_y]))),
...             torch.cartesian_prod(x, y))
True

`torch.meshgrid` is commonly used to produce a grid for
plotting.
>>> import matplotlib.pyplot as plt
>>> xs = torch.linspace(-5, 5, steps=100)
>>> ys = torch.linspace(-5, 5, steps=100)
>>> x, y = torch.meshgrid(xs, ys, indexing='xy')
>>> z = torch.sin(torch.sqrt(x * x + y * y))
>>> ax = plt.axes(projection='3d')
>>> ax.plot_surface(x.numpy(), y.numpy(), z.numpy())
>>> plt.show()
../_images/meshgrid.png

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源