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 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录