Template Function deepmd::session_get_vector

Function Documentation

template<typename VT>
void deepmd::session_get_vector(std::vector<VT> &o_vec, tensorflow::Session *session, const std::string name_, const std::string scope = "")

Get the vector of a tensor.

Parameters
  • o_vec[out] The output vector.

  • session[in] TensorFlow session.

  • name[in] The name of the tensor.

  • scope[in] The scope of the tensor.