TensorFlow 將給定的 type_value 轉(zhuǎn)換為 DType

tf.as_dtype

as_dtype ( type_value )

定義在：tensorflow/python/framework/dtypes.py

參見指南：構(gòu)建圖>張量類型

將給定的 type_value 轉(zhuǎn)換為 DType.

ARGS：

• type_value：可以轉(zhuǎn)換為 tf.DType 對象的值,這可能的英文一個(gè) tf.DType 對象,一個(gè) DataType 枚舉,一個(gè)字符串類型名稱或一個(gè) numpy.dtype.

返回：

對應(yīng)于 type_value 的 DType.

注意：

• TypeError：如果 type_value 不能轉(zhuǎn)換為 DType.