Function deepmd::session_get_dtype#
Defined in File commonTF.h
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.