首页 > 解决方案 > 如何在 C++ 中的 libtorch 中按第一维对张量进行排序?

问题描述

我有以下二维张量,我想按第一维排序:

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

标签: c++pytorchlibtorch

解决方案


推荐阅读