mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
36 lines
737 B
C
36 lines
737 B
C
#ifndef TH_TENSOR_INC
|
|
#define TH_TENSOR_INC
|
|
|
|
#include "THStorage.h"
|
|
#include "THTensorApply.h"
|
|
|
|
#define THTensor TH_CONCAT_3(TH,Real,Tensor)
|
|
#define THTensor_(NAME) TH_CONCAT_4(TH,Real,Tensor_,NAME)
|
|
|
|
/* basics */
|
|
#include "generic/THTensor.h"
|
|
#include "THGenerateAllTypes.h"
|
|
|
|
#include "generic/THTensorCopy.h"
|
|
#include "THGenerateAllTypes.h"
|
|
|
|
#include "THTensorMacros.h"
|
|
|
|
/* random numbers */
|
|
#include "generic/THTensorRandom.h"
|
|
#include "THGenerateAllTypes.h"
|
|
|
|
/* maths */
|
|
#include "generic/THTensorMath.h"
|
|
#include "THGenerateAllTypes.h"
|
|
|
|
/* convolutions */
|
|
#include "generic/THTensorConv.h"
|
|
#include "THGenerateAllTypes.h"
|
|
|
|
/* lapack support */
|
|
#include "generic/THTensorLapack.h"
|
|
#include "THGenerateFloatTypes.h"
|
|
|
|
#endif
|