PyTorch función torch.kthvalue
Manual de referencia de Pytorch torch
torch.kthvalueEs una función en PyTorch utilizada para devolver el k-ésimo valor más pequeño. Devuelve el k-ésimo valor mínimo y su índice en la dimensión especificada del tensor de entrada.
Definición de la función
torch.kthvalue(input, k, dim=None, keepdim=False)
Ejemplo de uso
Ejemplo
import torch
# Crear tensor
x = torch.tensor([3, 1, 2, 4])
# Devolver el segundo valor más pequeño y su índice
values, indices = torch.kthvalue(x, 2)
print(values)
print(indices)
# Crear tensor
x = torch.tensor([3, 1, 2, 4])
# Devolver el segundo valor más pequeño y su índice
values, indices = torch.kthvalue(x, 2)
print(values)
print(indices)
El resultado de salida es:
tensor(2) tensor(2)
Otras extensiones