Pytorch之torch.meshgrid()

說明：

　　torch.meshgrid()的功能是生成網格，可以用於生成坐標。

函數輸入:

　　輸入兩個數據類型相同的一維tensor

函數輸出：

       輸出兩個tensor（tensor行數為第一個輸入張量的元素個數，列數為第二個輸入張量的元素個數）

注意：

　　1）當兩個輸入tensor數據類型不同或維度不是一維時會報錯。

　　2）其中第一個輸出張量填充第一個輸入張量中的元素，各行元素相同；第二個輸出張量填充第二個輸入張量中的元素各列元素相同。<填充效果見實驗結果>

實驗驗證：

import torch
x = torch.tensor([1,2,3]) #定義以為張量x，作為行輸入張量
y = torch.tensor([5,6,7]) #定義以為張量y，作為行列入張量
resultx,resulty = torch.meshgrid(x,y)
print(x,y)
print(resultx)
print(resulty)

實驗結果（jupyter notebook）：　

tensor([1, 2, 3]) tensor([5, 6, 7])
tensor([[1, 1, 1],
        [2, 2, 2],
        [3, 3, 3]])
tensor([[5, 6, 7],
        [5, 6, 7],
        [5, 6, 7]])