1 #pragma once 2 3 #include <torch/csrc/autograd/variable.h> 4 #include <torch/csrc/python_headers.h> 5 #include <cstdint> 6 7 extern PyTypeObject THPSizeType; 8 9 #define THPSize_Check(obj) (Py_TYPE(obj) == &THPSizeType) 10 11 PyObject* THPSize_New(const torch::autograd::Variable& t); 12 PyObject* THPSize_NewFromSizes(int64_t dim, const int64_t* sizes); 13 PyObject* THPSize_NewFromSymSizes(const at::Tensor& t); 14 15 void THPSize_init(PyObject* module); 16