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.