c++ 输出 张量 shape
时间: 2024-04-23 09:28:53 浏览: 125
C语言 c++ 输出图形 数学
引用\[1\]:在C++中输出张量的形状可以通过以下代码实现:
```cpp
#include <iostream>
#include <vector>
int main() {
std::vector<int> shape = {1, 1, 3, 3}; // 假设张量的形状是\[1, 1, 3, 3\]
for (int i = 0; i < shape.size(); i++) {
std::cout << shape\[i\] << " ";
}
return 0;
}
```
这段代码使用了一个`std::vector`来存储张量的形状,然后通过遍历`shape`向量输出每个维度的大小。在这个例子中,输出的结果是`1 1 3 3`,表示张量的形状是一个4维张量,其中第一维大小为1,第二维大小为1,第三维大小为3,第四维大小为3。\[1\]
希望这个回答对您有帮助!
#### 引用[.reference_title]
- *1* [C++ STL vector的输出](https://blog.csdn.net/qq78442761/article/details/79682247)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^koosearch_v1,239^v3^insert_chatgpt"}} ] [.reference_item]
- *2* *3* [C++&TensorRT | 推理时动态shape](https://blog.csdn.net/weixin_43236007/article/details/128460888)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^koosearch_v1,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]
阅读全文