THTensor_getStoragePtr(tensor) 源码解析:获取张量数据存储指针
THTensor_getStoragePtr(tensor) 是一个宏定义,用于获取张量 (tensor) 的数据存储部分的指针。
具体实现如下:
#define THTensor_getStoragePtr(tensor) \
(((tensor)->storage == NULL) ? NULL : (tensor)->storage->data)
该宏首先判断张量的 storage 是否为 NULL。如果为 NULL,则返回 NULL;否则,返回 storage 中的数据指针。storage 是存储张量数据的容器,它是一个 THStorage 类型的指针,而 THStorage 结构体中有一个 data 字段,存储了数据的指针。因此,THTensor_getStoragePtr(tensor) 实际上返回的是张量关联的 storage 中存储数据的指针。
原文地址: https://www.cveoy.top/t/topic/n1dO 著作权归作者所有。请勿转载和采集!