Preguntas con Tag: libtorch

La libtorch es el frontend C++ de PyTorch, que ofrece una interacción de alto nivel para el uso de la investigación y producción de aprendizaje automático.

¿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});
c++ pytorch libtorch
techinnovator
3 años, 4 meses, 29 días