1
2
3
4
5
6
7
8
9
10
11
12
13
| #ifndef TH_SIZE_INC
| #define TH_SIZE_INC
|
| #include <TH/THGeneral.h>
| #include <stddef.h>
|
| // THTensor functions that would work on a THSize if we had such a class in C++,
| // i.e. THTensor functions that depend only on the shape of the tensor, not the type.
|
| TH_API int THSize_isSameSizeAs(const int64_t *sizeA, int64_t dimsA, const int64_t *sizeB, int64_t dimsB);
| TH_API ptrdiff_t THSize_nElement(int64_t dims, int64_t *size);
|
| #endif
|
|
