Fonction torch.bucketize de PyTorch

Pytorch torch 参考手册 Manuel de référence Pytorch torch

torch.bucketizeC'est une fonction PyTorch utilisée pour l'indexation par compartiments. Elle associe les valeurs d'entrée aux indices de compartiment correspondants, en effectuant une recherche binaire basée sur un tableau de limites. Elle est souvent utilisée pour discrétiser des valeurs continues ou pour des opérations de regroupement par intervalles.

Définition de fonction

torch.bucketize(input, boundaries, right=False, out_int32=False, **kwargs)

Exemple d'utilisation

Exemple

import torch

# Utilisation de base : indexation par compartiments
boundaries = torch.tensor([0, 1, 2, 3, 4])
values = torch.tensor([0.5, 1.5, 2.5, 3.5, 0.2, 2.0])

result = torch.bucketize(values, boundaries)
print("Limites :", boundaries)
print("Valeurs :", values)
print("Indices de compartiment :", result)
# Sortie : tensor([1, 2, 3, 4, 0, 2])

# right=True signifie que les bornes sont des intervalles fermés à droite
result_right = torch.bucketize(values, boundaries, right=True)
print("indices de compartiment right=True :", result_right)
# Sortie : tensor([0, 1, 2, 3, 0, 1])

# Entrée multidimensionnelle
values = torch.tensor([[0.5, 1.5], [2.5, 3.5]])
result = torch.bucketize(values, boundaries)
print("Résultat de l'entrée multidimensionnelle :", result)

Pytorch torch 参考手册 Manuel de référence Pytorch torch

Autres extensions