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(int dim, const int64_t* sizes); |
13 | PyObject* THPSize_NewFromSymSizes(const at::Tensor& t); |
14 | |
15 | void THPSize_init(PyObject* module); |
16 |