PyTorch torch.combinations Fonction

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

torch.combinationsC'est une fonction dans PyTorch utilisée pour calculer toutes les combinaisons de r éléments parmi les éléments du tenseur d'entrée. Elle renvoie toutes les combinaisons possibles de longueur r à partir du tenseur d'entrée.

Définition de fonction

torch.combinations(input, r=2, with_replacement=False)

Exemples d'utilisation

Exemple

import torch

# Calculer toutes les combinaisons de 2 éléments
x = torch.tensor([1, 2, 3, 4])
result = torch.combinations(x, r=2)
print("Entrée :", x)
print("combinaisons de 2 éléments :")
print(result)
# tensor([[1, 2],
#         [1, 3],
#         [1, 4],
#         [2, 3],
#         [2, 4],
#         [3, 4]])

# Combinaisons de 3 éléments
result3 = torch.combinations(x, r=3)
print("combinaisons de 3 éléments :")
print(result3)

# Combinaisons avec remise (with_replacement=True)
result_with_replacement = torch.combinations(x, r=2, with_replacement=True)
print("Combinaisons de 2 éléments avec remplacement :")
print(result_with_replacement)

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

Autres extensions