Function get_data_size
Defined in File quantize.h
Function Documentation
-
uint32_t get_data_size(const uint32_t *sizes, uint32_t num_of_dims, int data_type)
计算张量数据的总字节大小。
根据张量的各维尺寸和数据类型，返回对应的数据大小（以字节为单位）， 内部通过获取元素总数并乘以每个元素的字节数来实现
- 参数:
sizes -- [IN] 张量的维度数组（如 [1, 3, 224, 224]）
num_of_dims -- [IN] 张量的维度个数。
data_type -- [IN] 数据类型（如 PNNA_BUFFER_FORMAT_UINT8 等），由外部类型系统定义
- 返回:
uint32_t
返回张量在内存中占用的总字节数
若输入参数无效（如 sizes 为 NULL），则返回 0