THCudaTensor_getDevice() 函数解析：获取 THCudaTensor 设备编号

THCudaTensor_getDevice() 函数解析：获取 THCudaTensor 设备编号

该函数用于获取 THCudaTensor 在哪个 GPU 设备上进行计算。

函数源码:

THC_API int THCudaTensor_getDevice(THCState* state, const THCudaTensor* tensor) {
  return THCStorage_getDevice(state, tensor->storage);
}

函数分析:

该函数实际上是通过调用 THCStorage_getDevice() 函数来获取 THCudaTensor 所关联的 THCudaStorage 对象所在的设备编号。因为 THCudaTensor 对象中存储的是 THCudaStorage 对象的指针，所以可以直接通过获取 THCudaStorage 对象的设备编号来获取 THCudaTensor 所在的设备编号。

函数参数中的 state 参数用于指定 THCudaTensor 所关联的 THCState 对象。tensor 参数指定需要获取设备编号的 THCudaTensor 对象。

函数返回值为一个整型数，表示 THCudaTensor 所在的设备编号。