[ Pytorch ] meshgrid, cumsum란?

meshgrid

     - 각각의 원소가 scalar 또는 1 dimensional vector인 N개의 tensor를 받아, N개의 N dimensional grid를 만듭니다.
        i번째 grid는 i번째 input을 다른 input으로 정의된 차원만큼 확장한 것이다.

import torch

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

grid_x, grid_y = torch.meshgrid(x, y, indexing = "xy")
print(grid_x)
print(grid_y)

>> tensor([[1, 2, 3],
           [1, 2, 3],
           [1, 2, 3]])
        
>> tensor([[4, 4, 4],
           [5, 5, 5],
           [6, 6, 6]])
           
grid_x, grid_y = torch.meshgrid(x, y, indexing = "ij")
print(grid_x)
print(grid_y)

>> tensor([[1, 1, 1],
          [2, 2, 2],
          [3, 3, 3]])
>> tensor([[4, 5, 6],
          [4, 5, 6],
          [4, 5, 6]])

cumsum

       - 배열에서 주어진 축에 따라 누적되는 원소들의 누적 합을 계산하는 함수

       

참고

https://pytorch.org/docs/stable/generated/torch.cumsum.html

https://pytorch.org/docs/stable/generated/torch.meshgrid.html

torch.meshgrid — PyTorch 1.13 documentation