Функция PyTorch torch.combinations
Справочное руководство PyTorch torch
torch.combinationsЭто функция в PyTorch, используемая для вычисления всех r-элементных комбинаций элементов входного тензора. Она возвращает все возможные комбинации длины r из входного тензора.
Определение функции
torch.combinations(input, r=2, with_replacement=False)
Примеры использования
Пример
import torch
# Вычислить все 2-элементные комбинации
x = torch.tensor([1, 2, 3, 4])
result = torch.combinations(x, r=2)
print("Вход:", x)
print("2-элементные комбинации:")
print(result)
# tensor([[1, 2],
# [1, 3],
# [1, 4],
# [2, 3],
# [2, 4],
# [3, 4]])
# 3-элементные комбинации
result3 = torch.combinations(x, r=3)
print("3-элементные комбинации:")
print(result3)
# Комбинации с повторением (with_replacement=True)
result_with_replacement = torch.combinations(x, r=2, with_replacement=True)
print("2-элементные комбинации с повторениями:")
print(result_with_replacement)
# Вычислить все 2-элементные комбинации
x = torch.tensor([1, 2, 3, 4])
result = torch.combinations(x, r=2)
print("Вход:", x)
print("2-элементные комбинации:")
print(result)
# tensor([[1, 2],
# [1, 3],
# [1, 4],
# [2, 3],
# [2, 4],
# [3, 4]])
# 3-элементные комбинации
result3 = torch.combinations(x, r=3)
print("3-элементные комбинации:")
print(result3)
# Комбинации с повторением (with_replacement=True)
result_with_replacement = torch.combinations(x, r=2, with_replacement=True)
print("2-элементные комбинации с повторениями:")
print(result_with_replacement)
Другие расширения