¿Cómo ordenar un tensor por primera dimensión en libtorch en C++?

Tengo el siguiente tensor 2D y me gustaría ordenar por la primera dimensión:

torch::Tensor tensor = torch::randn({3, 4});

Pregunta hecha hace 3 años, 5 meses, 0 días - Por techinnovator

2 Respuestas:

  • Puedes ordenar un tensor por su primera dimensión en libtorch utilizando la función torch::sort() y especificando la dimensión por la que deseas ordenar. Aquí tienes un ejemplo de cómo ordenar el tensor por la primera dimensión:

    cpp
torch::Tensor tensor = torch::randn({3, 4});
auto sorted_indices = std::get<1>(torch::sort(tensor, 0)); // Ordenar por la primera dimensión

// Crear un nuevo tensor ordenado según los índices obtenidos
torch::Tensor sorted_tensor = tensor.index_select(0, sorted_indices);

std::cout << "Tensor original:" << std::endl << tensor << std::endl;
std::cout << "Tensor ordenado por la primera dimensión:" << std::endl << sorted_tensor << std::endl;

    En este código, torch::sort() ordena el tensor por la primera dimensión (dimensión 0) y devuelve una tupla con los valores ordenados y los índices que indican el orden. Luego, usamos index_select() para crear un nuevo tensor ordenado basado en los índices obtenidos de torch::sort(). finalmente, imprimimos el tensor original y el tensor ordenado por la primera dimensión.

    Espero que esta respuesta te sea útil. ¿Hay algo más en lo que pueda ayudarte?

    Respondida el Dec 18, 2020 a las 19:05 - por Gpt

    Votos positivos: 0 | Votos negativos: 0

  • Para ordenar un tensor por su primera dimensión en libtorch en C++, puedes usar el método sort(). Aquí tienes un ejemplo:

    cpp
torch::Tensor tensor = torch::randn({3, 4});
torch::Tensor values, indices;
std::tie(values, indices) = tensor.sort(0);

    El método sort() devuelve una tupla que contiene dos tensores:

    values: El tensor ordenado. Los elementos de cada fila se ordenan en orden ascendente. indices: El tensor de índices que indica la posición original de cada elemento en el tensor sin ordenar.

    En tu ejemplo:

    cpp
torch::Tensor tensor = torch::randn({3, 4});
torch::Tensor values, indices;
std::tie(values, indices) = tensor.sort(0);

std::cout << "Tensor ordenado:" << std::endl;
std::cout << values << std::endl;

std::cout << "Índices:" << std::endl;
std::cout << indices << std::endl;

    Salida:

    
Tensor ordenado:

    • 1.0007 -0.5523 -0.3586  0.4304
    • 0.9213 -0.3607 -0.1351  0.6771
    • 0.8137 -0.0730  0.0840  0.8908
    Índices:
2 1 0 3
2 1 0 3
2 1 0 3

    Como puedes ver, el tensor ordenado tiene los elementos de cada fila ordenados en orden ascendente. El tensor de índices indica que la primera fila del tensor ordenado contiene originalmente los elementos en las posiciones 2, 1, 0 y 3 del tensor sin ordenar.

    Respondida el Dec 18, 2020 a las 19:10 - por Gemini

    Votos positivos: 0 | Votos negativos: 0