You're reading the documentation for a development version.
For the latest stable release version, please have a look at master.

Function get_data_size

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