[pytorch] Advanced indexing that supports all indexing features on PyTorch#1719

Merged

KexinFeng merged 15 commits intodeepjavalibrary:masterfrom KexinFeng:torch_index

Jun 28, 2022

+295-20

Commits on Jun 27, 2022

mixed index getter on pytorch draft

KexinFeng
authored and
frankfliu
committed
code cleaning

KexinFeng
authored and
frankfliu
committed
code cleaning 2

KexinFeng
authored and
frankfliu
committed
feed std::vector<> to tensor.index(ArrayRef<>)

KexinFeng
authored and
frankfliu
committed
bug fixed

KexinFeng
authored and
frankfliu
committed
Update api/src/main/java/ai/djl/ndarray/index/dim/NDIndexNone.java

KexinFeng
and
frankfliu
committed
bug fixed

KexinFeng
authored and
frankfliu
committed
bug fixed

KexinFeng
authored and
frankfliu
committed
Torch index type check: long, byte or boolean; restore testPick behaviour; The previous commit: PtNDManager.from() bug fixed.

KexinFeng
authored and
frankfliu
committed
Restore the get(NDArray, NDIndex)

KexinFeng
authored and
frankfliu
committed
change at::indexing to torch::indexing; testRuntimeOnly project(":engines:pytorch:pytorch-jni

KexinFeng
authored and
frankfliu
committed
Add :engines:pytorch:pytorch-jni

KexinFeng
authored and
frankfliu
committed
testIndexationUsesSpecificManager add manager checking into PtNDArrayIndexer get(NDArray, NDIndex)

KexinFeng
authored and
frankfliu
committed
Add NDIndeTest fix NDArray.get(index)

KexinFeng
authored and
frankfliu
committed
Format c++ code
frankfliu
committed