1. description

x_meshgrid,y_meshgrid=torch.meshgrid(x_tensor,y_tensor) 函数的功能是根据x_tensor,y_tensor来生成网格矩阵x_meshgrid,y_meshgrid,注意这里的x_tensor和y_tensor的方向和实际方向相反，具体如下描述

• excel:
  

2. pytorch

• pytorch

import torch
import torch.nn as nn

torch.set_printoptions(precision=3, sci_mode=False)

if __name__ == "__main__":
    run_code = 2
    x_len = 4
    y_len = 5
    x_tensor = torch.arange(x_len) + 1
    y_tensor = torch.arange(y_len) + 2
    x_meshgrid, y_meshgrid = torch.meshgrid(x_tensor, y_tensor)
    print(f"x_tensor=\n{x_tensor}")
    print(f"y_tensor=\n{y_tensor}")
    print(f"x_meshgrid=\n{x_meshgrid}")
    print(f"y_meshgrid=\n{y_meshgrid}")

• result:

x_tensor=
tensor([1, 2, 3, 4])
y_tensor=
tensor([2, 3, 4, 5, 6])
x_meshgrid=
tensor([[1, 1, 1, 1, 1],
        [2, 2, 2, 2, 2],
        [3, 3, 3, 3, 3],
        [4, 4, 4, 4, 4]])
y_meshgrid=
tensor([[2, 3, 4, 5, 6],
        [2, 3, 4, 5, 6],
        [2, 3, 4, 5, 6],
        [2, 3, 4, 5, 6]])