что делает 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 возвращает уплощенный одномерный массив тензора.

Надеждаэто помогает