libtorch - Torch::zeros 创建

问题描述

我想torch::zeros基于来自std::vector<int64_t> vec. 显然不可能通过创建张量torch::zeros(vec)，因为它只接受像{a,b,c}or这样的数组的右值c10::ArrayRef。由于我的尺寸长度发生了vec变化，我无法使用{a,b,c}，我必须遵循第二种方法。知道如何将 a 转换std:vector为c10::ArrayRef吗？

我也尝试通过传递一个数组int* vec = &vec[0];，仍然得到 api 的类型转换错误。另外，我尝试创建一个arrayRefvia auto dims = c10::ArrayRef<int64_t>(vec.size());，但 torch 不喜欢它的语法。

标签： libtorch

libtorch - Torch::zeros 创建

问题描述

解决方案

推荐阅读