1 Star 0 Fork 0

Velcon-Zheng/cutorch

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
Tensor.c 783 Bytes
一键复制 编辑 原始数据 按行查看 历史
#include "torch/utils.h"
#include "THC.h"
#include "THFile.h"
#include "luaT.h"
#define torch_Storage_(NAME) TH_CONCAT_4(torch_,CReal,Storage_,NAME)
#define torch_Storage TH_CONCAT_STRING_3(torch.,CReal,Storage)
#define torch_Tensor_(NAME) TH_CONCAT_4(torch_,CReal,Tensor_,NAME)
#define torch_Tensor TH_CONCAT_STRING_3(torch.,CReal,Tensor)
#define cutorch_Tensor_(NAME) TH_CONCAT_4(cutorch_,CReal,Tensor_,NAME)
#define cutorch_TensorCopy_(NAME) TH_CONCAT_4(cutorch_,Real,TensorCopy_,NAME)
// generate the torch types -- we could also do this via THGenerateAllTypes,
// but this allows us to be self contained.
#define FORCE_TH_HALF
#include "generic/CTensorCopy.c"
#include "THCGenerateAllTypes.h"
#undef FORCE_TH_HALF
#include "generic/CTensor.c"
#include "THCGenerateAllTypes.h"
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/Velcon-Zheng/cutorch.git
git@gitee.com:Velcon-Zheng/cutorch.git
Velcon-Zheng
cutorch
cutorch
master

搜索帮助