What is THCudaTensor_data corresponding function in Aten

C++
1

I’m converting a Pytorch 0.4 version model to Pytorch 1.3. I failed to build some self defined .c files, because Pytorch deprecated THC, and use ATen as CUDA lib. Can anyone tell me what is the corresponding THCudaTensor_data function in ATen? Or how should I modify the C files?

2

probably tensor.data_ptr()

3

Hi @SimonW, thanks for your reply. It seems right, both functions return the address of given tensor. However, I want to use it in c++, should I use something like at::tensor.data_ptr()?

4

yeah, I actually meant the cpp code. :slight_smile:

1 Like
5

Thank you! will give a try.