tf.debugging.assert_same_float_dtype
Проверка и возвращение типа float на основе tensors и dtype.
tf.debugging.assert_same_float_dtype(
tensors=None, dtype=None
)
Для операций, таких как умножение матриц, входные данные и веса должны иметь один и тот же тип float. Эта функция проверяет, что все tensors имеют один и тот же тип, проверяет, что тип является dtype (если указан), и возвращает тип. Тип должен быть плавающей точкой. Если ни tensors, ни dtype не указаны, функция вернет dtypes.float32.
| Аргументы | |
|---|---|
tensors | Тензоры входных значений. Может включать None элементы, которые будут проигнорированы. |
dtype | Ожидаемый тип. |
| Возвращает | |
|---|---|
| Проверенный тип. |
| Исключения | |
|---|---|
ValueError | если ни tensors, ни dtype не указаны, или результат не является типом float, или общий тип входных данных не является типом с плавающей точкой. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/debugging/assert_same_float_dtype