THCudaTensor_data（通常是THC）做什么？

Question

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

1. THC的实施方式（如果可能的话，我会很详细地介绍一些细节）？
2. THCudaTensor_data（THC_state，THCudaTensor *）有什么作用？ （从代码中使用它的方式来看，似乎将其用于将pytorch的张量转换为cuda中的数组。如果是这种情况，那么函数将保留所有元素和数组的长度吗？）

Answer 1

我仍然不确定THCudaTensor_data的内部工作原理，但是让我感到震惊的行为是：对于n维张量，THCudaTensor_data返回张紧的一维数组。

希望这会有所帮助