bmruntime中bm_tensor_t 维度打印
时间: 2024-05-04 13:20:23 浏览: 179
在bmruntime中,可以使用`bm_tensor_shape()`函数来获取`bm_tensor_t`的维度信息,然后使用循环打印出来。以下是一个示例代码:
```c
#include "bm_runtime.h"
#include <stdio.h>
void print_tensor_shape(const bm_tensor_t* tensor) {
int ndim = bm_tensor_ndim(tensor);
const int* shape = bm_tensor_shape(tensor);
printf("Tensor shape: [");
for (int i = 0; i < ndim; ++i) {
printf("%d", shape[i]);
if (i < ndim - 1) {
printf(", ");
}
}
printf("]\n");
}
```
在这个函数中,我们首先使用`bm_tensor_ndim()`函数获取`bm_tensor_t`的维度数目,并使用`bm_tensor_shape()`函数获取`bm_tensor_t`的维度信息。然后我们使用循环遍历维度信息并打印出来,注意最后一个维度后面不需要打印逗号。使用示例如下:
```c
bm_tensor_t* tensor = ...; // 创建一个bm_tensor_t对象
print_tensor_shape(tensor); // 打印tensor的维度信息
```
阅读全文