torch.meshgrid()

示例文件 test.py

import torch

cor1 = torch.arange(5).float()
cor2 = torch.arange(4).float()
print(cor1)
print(cor2)

X, Y = torch.meshgrid(cor1, cor2)
print(X)
print(Y)

终端命令行及运行结果

<user>python test.py
tensor([0., 1., 2., 3., 4.])
tensor([0., 1., 2., 3.])
tensor([[0., 0., 0., 0.],
        [1., 1., 1., 1.],
        [2., 2., 2., 2.],
        [3., 3., 3., 3.],
        [4., 4., 4., 4.]])
tensor([[0., 1., 2., 3.],
        [0., 1., 2., 3.],
        [0., 1., 2., 3.],
        [0., 1., 2., 3.],
        [0., 1., 2., 3.]])

从运算结果来看：

返回两个tensor，size就是 x.size * y.size (行数是x的个数，列数是y的个数)。

可以想象成这样：

tensor([[0., 1., 2., 3.],
        [1.,  .,  .,  .],
        [2.,  .,  .,  .],
        [3.,  .,  .,  .],
        [4.,  .,  .,  .]])

返回的第一个tensor是竖着的列往右平移，返回的第二个tensor是横着的行往下平移。

注意：
1）两个输入参数都是一维张量。
2）两个输入参数的张量类型要相同。