Función torch.amin de PyTorch

Pytorch torch 参考手册Manual de referencia de PyTorch torch

torch.aminEs una función de PyTorch que se utiliza para devolver el valor mínimo de un tensor a lo largo de una dimensión especificada.

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

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

# Devuelve el valor mínimo de todos los elementos
print("Mínimo global:", torch.amin(x))

# Valor mínimo a lo largo de dim=0
print("Mínimo en dim=0:", torch.amin(x, dim=0))

# Valor mínimo a lo largo de dim=1
print("Mínimo en dim=1:", torch.amin(x, dim=1))

El resultado de salida es:

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

Pytorch torch 参考手册Manual de referencia de PyTorch torch

Otras extensiones