tf.debugging.assert_same_float_dtype

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.

Args
tensors Tensors of input values. Can include None elements, which will be ignored.
dtype Expected type.
Returns
Validated type.
Raises
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.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/debugging/assert_same_float_dtype