THTensor_getStoragePtr(tensor)是一个宏定义,其作用是获取tensor的数据存储部分的指针。

具体实现如下:

#define THTensor_getStoragePtr(tensor) \
  (((tensor)->storage == NULL) ? NULL : (tensor)->storage->data)

该宏定义首先判断tensor的storage是否为NULL,如果为NULL则返回NULL,否则返回storage的数据指针。其中,storage是存储tensor数据的容器,它是THStorage类型的指针,而THStorage结构体中有一个data字段,存储了数据的指针。因此,THTensor_getStoragePtr(tensor)实际上返回的是tensor所关联的storage中存储数据的指针。

THTensor_getStoragePtrtensor源码分析

原文地址: https://www.cveoy.top/t/topic/e15j 著作权归作者所有。请勿转载和采集!

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