PyTorch torch.combinations Funktion
PyTorch torch-Referenzhandbuch
torch.combinationsEs ist eine Funktion in PyTorch, die verwendet wird, um alle r-elementigen Kombinationen der Elemente eines Eingabe-Tensors zu berechnen. Sie gibt alle möglichen Kombinationen der Länge r aus dem Eingabe-Tensor zurück.
Funktionsdefinition
torch.combinations(input, r=2, with_replacement=False)
Anwendungsbeispiele
Beispiel
import torch
# Alle 2-elementigen Kombinationen berechnen
x = torch.tensor([1, 2, 3, 4])
result = torch.combinations(x, r=2)
print("Eingabe:", x)
print("2-elementige Kombinationen:")
print(result)
# tensor([[1, 2],
# [1, 3],
# [1, 4],
# [2, 3],
# [2, 4],
# [3, 4]])
# 3-elementige Kombinationen
result3 = torch.combinations(x, r=3)
print("3-elementige Kombinationen:")
print(result3)
# Kombinationen mit Zurücklegen (with_replacement=True)
result_with_replacement = torch.combinations(x, r=2, with_replacement=True)
print("2-elementige Kombinationen mit Zurücklegen:")
print(result_with_replacement)
# Alle 2-elementigen Kombinationen berechnen
x = torch.tensor([1, 2, 3, 4])
result = torch.combinations(x, r=2)
print("Eingabe:", x)
print("2-elementige Kombinationen:")
print(result)
# tensor([[1, 2],
# [1, 3],
# [1, 4],
# [2, 3],
# [2, 4],
# [3, 4]])
# 3-elementige Kombinationen
result3 = torch.combinations(x, r=3)
print("3-elementige Kombinationen:")
print(result3)
# Kombinationen mit Zurücklegen (with_replacement=True)
result_with_replacement = torch.combinations(x, r=2, with_replacement=True)
print("2-elementige Kombinationen mit Zurücklegen:")
print(result_with_replacement)
Andere Erweiterungen