tf.dtypes.as_dtype

View source on GitHub

Converts the given type_value to a DType.

tf.dtypes.as_dtype(
    type_value
)

Args:

• type_value: A value that can be converted to a tf.DType object. This may currently be a tf.DType object, a DataType enum, a string type name, or a numpy.dtype.

Returns:

A DType corresponding to type_value.

Raises:

• TypeError: If type_value cannot be converted to a DType.