torch-sys 0.20.0

Low-level FFI bindings for the PyTorch C++ api (libtorch).
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

#include "torch_python.h"

bool thp_variable_check(PyObject* pyobj) {
  PROTECT(
      return THPVariable_Check(pyobj);
  )
  return false;
}

PyObject* thp_variable_wrap(tensor t) {
  PROTECT(
      return THPVariable_Wrap(*t);
  )
  return nullptr;
}

tensor thp_variable_unpack(PyObject* pyobj) {
  PROTECT(
      return new torch::Tensor(THPVariable_Unpack(pyobj));
  )
  return nullptr;
}