PyTorch torch.amin-Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.aminEs ist eine Funktion in PyTorch, die verwendet wird, um den Minimalwert eines Tensors entlang einer angegebenen Dimension zurückzugeben.

Funktionsdefinition

torch.amin(input, dim, keepdim=False)

Anwendungsbeispiel

Beispiel

import torch

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

# Minimalwert aller Elemente zurückgeben
print("Globales Minimum:", torch.amin(x))

# Minimalwert entlang dim=0
print("Minimum für dim=0:", torch.amin(x, dim=0))

# Minimalwert entlang dim=1
print("Minimum für dim=1:", torch.amin(x, dim=1))

Das Ausgabeergebnis ist:

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

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Andere Erweiterungen