“Pytorch Tensor Argmax” Kod odpowiedzi

Pytorch Tensor Argmax

values, indices = tensor.max(0)
values, indices = torch.max(tensor, 0)