c++ 一维 tensor 长度
时间: 2023-11-30 19:02:25 浏览: 139
C ++ Cuda Tensor惰性计算库-C/C++开发
根据引用\[2\]中的描述,如果使用torch::tensor创建一个一维tensor,可以使用std::vector<float>作为输入,并将其转换为torch::Tensor对象。例如,如果我们有一个名为scales的std::vector<float>,我们可以使用以下代码将其转换为一维tensor:
std::vector<float> scales = {1.0, 2.0, 3.0, 4.0, 5.0};
torch::Tensor scales_tensor = torch::tensor(scales);
这样,scales_tensor就是一个长度为5的一维tensor。
#### 引用[.reference_title]
- *1* *2* *3* [c++ 部署libtorch 时对Tensor块的常用操作API (例如vector转换为torch、squeeze、select、select_indxe、max等)](https://blog.csdn.net/weixin_45209433/article/details/103495558)[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 ]
阅读全文