Function deepmd::session_get_dtype

Function Documentation

int deepmd::session_get_dtype(tensorflow::Session *session, const std::string name, const std::string scope = "")

Get the type of a tensor.

Parameters:

  • session – [in] TensorFlow session.

  • name – [in] The name of the tensor.

  • scope – [in] The scope of the tensor.

Returns:

The type of the tensor as int.