Función torch.min de PyTorch
Manual de referencia de Pytorch torch
torch.minEs la función utilizada en PyTorch para calcular el valor mínimo de los tensores.
Definición de la función
torch.min(input, dim, keepdim, out)
Ejemplo de uso
Ejemplo
import torch
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
print("Mínimo global:", torch.min(x))
print("Mínimo en dim=0:", torch.min(x, dim=0))
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
print("Mínimo global:", torch.min(x))
print("Mínimo en dim=0:", torch.min(x, dim=0))
Otras extensiones