#libtorch
Вопрос:
Я хочу создать а torch::zeros на основе измерения, которое происходит из std::vector<int64_t> vec А. По-видимому , невозможно создать тензор с помощью torch::zeros(vec) , так как он принимает только значение r массивов, таких как {a,b,c} или c10::ArrayRef . Поскольку длина моего размера vec меняется, я не могу использовать {a,b,c} , и мне приходится следовать второму подходу. Есть идеи, как я могу преобразовать a std:vector в c10::ArrayRef ?
Я также попытался передать массив int* vec = amp;vec[0]; , но все равно получил ошибку преобразования типов для api. Кроме того, я попытался создать arrayRef via auto dims = c10::ArrayRef<int64_t>(vec.size()); , но мне не понравился его синтаксис.