THCudaTensor_data （和一般的 THC ）做什么？答案

【问题标题】：what does THCudaTensor_data ( and THC in general ) do?THCudaTensor_data （和一般的 THC ）做什么？
【发布时间】：2019-03-12 20:45:50
【问题描述】：

我正在检查的程序使用 pytorch 加载权重，并使用 cuda 代码对权重进行计算。我对 THC 库的理解是张量是如何在 pytorch 的后端实现的（也许还有火炬？）。

THC 是如何实现的（如果可能的话，我真的很想了解一些细节）？
THCudaTensor_data(THC_state, THCudaTensor*) 有什么作用？（从它在代码中的使用方式来看，它似乎是用于将pytorch的张量转换为cuda中的数组。如果是这种情况，那么该函数会保留所有元素和数组的长度吗？）李>

【问题讨论】：

【解决方案1】：

我仍然不确定 THCudaTensor_data 的内部工作原理，但让我感到困惑的是：对于 n 维张量，THCudaTensor_data 返回张量的扁平一维数组。

希望对你有帮助

【讨论】：