Función PyTorch torch.combinations

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.combinationsEs una función de PyTorch que se utiliza para calcular todas las combinaciones de r elementos del tensor de entrada. Devuelve todas las combinaciones posibles de longitud r en el tensor de entrada.

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

# Calcular todas las combinaciones de 2 elementos
x = torch.tensor([1, 2, 3, 4])
result = torch.combinations(x, r=2)
print("Entrada:", x)
print("Combinaciones de 2 elementos:")
print(result)
# tensor([[1, 2],
#         [1, 3],
#         [1, 4],
#         [2, 3],
#         [2, 4],
#         [3, 4]])

# Combinaciones de 3 elementos
result3 = torch.combinations(x, r=3)
print("Combinaciones de 3 elementos:")
print(result3)

# Combinaciones con reemplazo (with_replacement=True)
result_with_replacement = torch.combinations(x, r=2, with_replacement=True)
print("Combinaciones de 2 elementos con reemplazo:")
print(result_with_replacement)

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones