Получение верхних K значений тензора с использованием TensorFlow C API - PullRequest
0 голосов
/ 03 августа 2020

У меня есть два вектора двойных значений в C ++: std::vector<double> vec_a; std::vector<double> vec_b;. Я хочу произвести попарное умножение этих двух векторов, чтобы получить K^2 значений; а затем возьмите верхние значения K из этого продукта.

Поскольку попарное умножение - дорогостоящая операция, я хочу проделать это на GPU. И также необходимо использовать API TensorFlow C. Документация по TensorFlow C API очень ограничена. Может ли кто-нибудь помочь мне с написанием вышеуказанного кода?

...