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