文件名称：AdvancedIndexing-PyTorch:（批次）PyTorch的高级索引
文件大小：14KB
文件格式：ZIP
更新时间：2024-04-04 14:59:27
nlp machine-learning natural-language-processing neural-network pytorch
AdvancedIndexing-PyTorch （批量）PyTorch的高级索引。 torch_index软件包设计用于在PyTorch张量上执行高级索引。 除了支持基本索引方法（矢量索引，外部索引，numpy样式索引，pytorch样式索引）之外，它还支持批量索引。 也就是说，张量的索引可能在不同的批处理索引中有所不同。 榜样胜过戒律！ import torch from torch_index import batch from torch_index import btindex a = torch . rand ( 2 , 3 , 6 , 7 , 8 ) blist = torch . tensor ([ [[ 0 , 1 ], [ 0 , 3 ], [ 0 , 5 ]], [[ 1 , 2 ], [ 1 , 3 ], [ 1 , 5 ]] ]) asse
【文件预览】：
AdvancedIndexing-PyTorch-master
----.vim-template:.py(254B)
----LICENSE(1KB)
----README.md(5KB)
----torch_index()
--------batched_functional.py(11KB)
--------utils.py(6KB)
--------__init__.py(2KB)
--------torch_utils.py(1KB)
--------basic_functional.py(3KB)
----tests()
--------test_batched.py(4KB)
--------test_basic.py(3KB)
----.gitignore(1KB)