文件名称：torch_indexing:用于高级pytorch张量索引的辅助库
文件大小：7KB
文件格式：ZIP
更新时间：2024-05-07 15:23:34
Cuda
用于高级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
【文件预览】：
torch_indexing-master
----torch_indexing()
--------src()
--------__init__.py(40B)
----CMakeLists.txt(2KB)
----test()
--------test_batch_sort.py(683B)
----README.md(1KB)
----csrc()
--------batch_sort.cu(3KB)
--------batch_sample.cu(4KB)
----.gitignore(24B)