[PYTORCH] torch.max 함수 설명
torch.max torch.max(input) -> Tensor torch.max(input, dim, keepdim=False, *, out=None) -> tuple (max, max_indices) torch.max 함수는 텐서에서 최대값을 구하는 함수이다. import torch import torch.nn as nn data = torch.randn((5, 5)) print(data) print(torch.max(data)) #### # tensor([[ 1.1328, -0.0392, -0.7076, 0.5610, 0.8010], # [-0.0898, -1.4467, -0.7285, -0.1195, -2.1070], # [ 0.4547, 1.7739, 0.1664, -1.0242, 0.0474..
[파이토치] torch.argmax 함수
torch.argmax torch.argmax(input) → LongTensor torch.argmax(input, dim, keepdim=False) → LongTensor 이 함수는 input tensor에 있는 모든 element들 중에서 가장 큰 값을 가지는 공간의 인덱스 번호를 반환하는 함수이다. a = torch.randn(4, 4) print(a) output = torch.argmax(a) print(output) #### output ##### ''' tensor([[-0.5014, -0.1785, 0.2534, 0.7167], [-0.7887, 1.0920, 0.5385, -1.1797], [-1.0129, 0.2337, 0.5757, 0.9139], [ 1.4672, -1.0605,..
[파이토치] torch.mm 함수
torch.mm torch.mm(input, mat2, *, out=None) → Tensor mm은 input과 mat2에 대해서 matrix multiplication을 수행하는 함수이다. 이 함수는 broadcast되지 않는 것이 특징이다. 만약 broadcasting된 것을 원한다면 torch.matul 함수를 사용하여아 한다. torch.mm( torch.tensor([[2, 2], [2, 2]]), torch.tensor([[2, 2], [2, 2]]) ) #### output #### ''' tensor([[8, 8], [8, 8]]) ''' 간단하게 torch를 import 하고 코드를 돌려보면 mm의 동작을 살펴볼 수 있다.
