Función torch.fmax de PyTorch

Manual de referencia de Pytorch torch

torch.fmaxEs la función en PyTorch para tomar el máximo elemento a elemento (ignorando NaN). Cuando la entrada contiene NaN, torch.maximum devuelve NaN, mientras que torch.fmax ignora NaN y devuelve valores no NaN.

Definición de funciones

torch.fmax(input, other)

Ejemplo de uso

Ejemplos

import torch

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

# Tomar el máximo por elemento (ignorando NaN)
result = torch.fmax(x, y)
print(result)

El resultado de salida es:

tensor([3., 5., 4.])

Manual de referencia de Pytorch torch

Otras extensiones