Pytorch Module.cpp

pytorch/torch/csrc/Module.cpp

Partly

#if PY_MAJOR_VERSION == 2
  ASSERT_TRUE(module = Py_InitModule("torch._C", methods.data()));
#else
  static struct PyModuleDef torchmodule = {
     PyModuleDef_HEAD_INIT,
     "torch._C",
     nullptr,
     -1,
     methods.data()
  };
  ASSERT_TRUE(module = PyModule_Create(&torchmodule));
#endif
  ASSERT_TRUE(THPWrapper_init(module));
  ASSERT_TRUE(THPGenerator_init(module));
  ASSERT_TRUE(THPException_init(module));
  THPSize_init(module);
  THPDtype_init(module);
  THPDTypeInfo_init(module);

Full

#include <torch/csrc/python_headers.h>
#include <sys/types.h>

#ifndef _MSC_VER
#i

猜你喜欢

转载自blog.csdn.net/tony2278/article/details/105071775