Missed TensorFlow World? Check out the recap. Learn more

tf.dtypes.as_dtype

TensorFlow 2 version View source on GitHub

Converts the given type_value to a DType.

Aliases:

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.