max, gather

<max> 

https://pytorch.org/docs/stable/generated/torch.max.html 의 내용이며, 각 인자의 설명 해석은 틀릴수 있음에 유의

 

torch.max(input)

- input tensor의 모든 요소중 최대값을 반환

 

사용 예제

a = torch.randn(1,3)

torch.max(a)
또는
a.max()

 

 

torch.max(input, dim, keepdim=False, *, out=None)

- 주어진 dim의 차원을 갖는 input이라는 텐서의 각 열(row)에서의 최대값과 해당 인덱스를 튜플로 반환 (values, indices)

- keepdim이 True라면, 각 차원의 size가 1인 경우를 제외하고 output 텐서는 input과 동일한 크기를 같는다

- keepdim이 False라면, squeeze 되어 차원은 1차원 축소된다

 

사용 예제

a = torch.randn(4,4)

torch.max(a,1)
또는
a.max(1)

 

 

 

 

 

<gather>

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

 

torch.gather(input, dim, index, *, sparse_grad=False, out=None)

- 특정 축을 따라 값을 추출한다

- dim : 기준 축 

- index : 추출할 인덱스들

 

주의할 점은 input과 index의 dimension이 동일해야함.

만약 input이 4x10x15 이고, dim =0이면 index는 Nx10x15

즉 dim인 부분을 제외하고 나머지 차원은 동일해야함

 

사용 예제

t = torch.tensor([[1, 2], [3, 4]])

torch.gather(t, 1, torch.tensor([[0, 0], [1, 0]]))
또는
t.gather(1, torch.tensor([[0, 0], [1, 0]]))

torch.gather(t, 0, torch.tensor([[0, 0], [1, 0]]))
또는
t.gather(0, torch.tensor([[0, 0], [1, 0]]))