Función PyTorch torch.fmin

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

torch.fminEs una función de PyTorch para obtener el mínimo elemento por elemento (ignorando NaN). Cuando la entrada contiene NaN, torch.minimum devuelve NaN, mientras que torch.fmin ignora NaN y devuelve el valor no NaN.

Definición de la función

torch.fmin(input, other)

Ejemplo de uso

Ejemplo

import torch

# Crear un tensor que contiene NaN
x = torch.tensor([3.0, float('nan'), 4.0])
y = torch.tensor([1.0, 5.0, 2.0])

# Obtener el mínimo elemento por elemento (ignorando NaN)
result = torch.fmin(x, y)
print(result)

El resultado de salida es:

tensor([1., 5., 2.])

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

Otras extensiones