Función torch.lt de PyTorch


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

torch.ltEs una función de PyTorch utilizada para la comparación elemento a elemento de menor que.

Definición de la función

torch.lt(input, other, out)

Ejemplo de uso

Ejemplo

import torch

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

result = torch.lt(a, b)

print(result)

El resultado de salida es:

tensor([ True, False, False, False])

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

Otras extensiones