PyTorch torch.argmin-Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.argminIst eine Funktion in PyTorch, die verwendet wird, um die Indizes der Minimalwerte entlang einer Dimension zurückzugeben.

Funktionsdefinition

torch.argmin(input, dim, keepdim)

Anwendungsbeispiele

Beispiel

import torch

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

print("Globaler Minimalindex:", torch.argmin(x))
print("dim=1 Minimalindex:", torch.argmin(x, dim=1))

Das Ausgabeergebnis ist:

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

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Andere Erweiterungen