torch.meshgrid()

• 2020 年 4 月 8 日
• 筆記

x1 ,y1 = torch.meshgrid(x,y) 參數是兩個，第一個參數我們假設是x，第二個參數假設就是y 輸出的是兩個tensor，size就是x.size * y.size（行數是x的個數，列數是y的個數） 具體輸出看下面

注意：兩個參數的數據類型要相同，要麼都是float，要麼都是int，否則會報錯。