Fonction torch.amin de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.aminC'est une fonction de PyTorch utilisée pour renvoyer la valeur minimale d'un tenseur le long d'une dimension spécifiée.

Définition de fonction

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

Exemple d'utilisation

Exemple

import torch

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

# Renvoyer la valeur minimale de tous les éléments
print("Minimum global :", torch.amin(x))

# Valeur minimale le long de dim=0
print("minimum pour dim=0 :", torch.amin(x, dim=0))

# Valeur minimale le long de dim=1
print("minimum pour dim=1 :", torch.amin(x, dim=1))

Le résultat de sortie est:

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

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions