mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
4518793aa2
Implements basic and advanced indexing using ATen tensors/variables. Basic indexing is translated at the Python-binding level (python_variable_indexing.cpp) to slice/squeeze/unsqueeze/select calls. Advanced indexing is implemented in ATen in terms of take() and put() calls.
13 lines
426 B
C
13 lines
426 B
C
#pragma once
|
|
|
|
#include <Python.h>
|
|
|
|
// https://bugsfiles.kde.org/attachment.cgi?id=61186
|
|
#if PY_VERSION_HEX >= 0x03020000
|
|
#define THPUtils_parseSlice(SLICE, LEN, START, STOP, LENGTH, STEP) \
|
|
(PySlice_GetIndicesEx(SLICE, LEN, START, STOP, LENGTH, STEP) == 0)
|
|
#else
|
|
#define THPUtils_parseSlice(SLICE, LEN, START, STOP, LENGTH, STEP) \
|
|
(PySlice_GetIndicesEx((PySliceObject*)SLICE, LEN, START, STOP, LENGTH, STEP) == 0)
|
|
#endif
|