mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
14 lines
288 B
C
14 lines
288 B
C
#ifndef TORCH_UTILS_INC
|
|
#define TORCH_UTILS_INC
|
|
|
|
#include "luaT.h"
|
|
#include "TH.h"
|
|
|
|
THLongStorage* torch_checklongargs(lua_State *L, int index);
|
|
int torch_islongargs(lua_State *L, int index);
|
|
|
|
void torch_setdefaulttensorid(const void* id);
|
|
const void* torch_getdefaulttensorid();
|
|
|
|
#endif
|