Function deepmd::session_get_dtype

• Defined in File common.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.