-
torch_indexing：一个辅助库，旨在支持PyTorch张量的高级索引功能
资源介绍
用于高级pytorch张量索引的辅助库
职能
当前实现的功能：
批量排序（CPU + CUDA）：批量排序一维张量
建造
我的开发环境：
pytorch：1.8.0 + cu111（通过pip安装）
CUDA：11.2
nvcc：11.2
推力：1.12.0
幼崽：1.12.0
cmake：3.18.4
编译示例：
# configure
cmake .. \
-DCMAKE_PREFIX_PATH= ` python -c ' import torch;print(torch.utils.cmake_prefix_path) ' ` \
-DPYTHON_INCLUDE_DIR= $( python -c " from distutils.sysconfig import get_python_inc; print(get_python_inc()) " ) \
-D