ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf.debugging.assert_same_float_dtype

TensorFlow 2 version View source on GitHub

Validate and return float type based on tensors and dtype.

For ops such as matrix multiplication, inputs and weights must be of the same float type. This function validates that all tensors are the same type, validates that type is dtype (if supplied), and returns the type. Type must be a floating point type. If neither tensors nor dtype is supplied, the function will return dtypes.float32.

tensors Tensors of input values. Can include None elements, which will be ignored.
dtype Expected type.

Validated type.

ValueError if neither tensors nor dtype is supplied, or result is not float, or the common type of the inputs is not a floating point type.