PyTorch torch.argmax Funktion

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

torch.argmaxEs ist eine Funktion in PyTorch, die verwendet wird, um den Index des Maximalwerts entlang einer Dimension zurückzugeben.

Funktionsdefinition

torch.argmax(input, dim, keepdim)

Anwendungsbeispiel

Beispiel

import torch

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

# Index des globalen Maximalwerts
print("Globaler Maximalindex:", torch.argmax(x))

# Index des Maximalwerts entlang dim=0
print("dim=0 Maximalindex:", torch.argmax(x, dim=0))

# Index des Maximalwerts entlang dim=1
print("dim=1 Maximalindex:", torch.argmax(x, dim=1))

Das Ausgabeergebnis ist:

全局最大索引: tensor(4)
dim=0 最大索引: tensor([1, 0, 1])
dim=1 最大索引: tensor([1, 0])

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

Andere Erweiterungen