tf.debugging.assert_same_float_dtype
Проверяет и возвращает тип с плавающей точкой на основе tensors и dtype.
tf.debugging.assert_same_float_dtype(
tensors=None, dtype=None
)
Для таких операций, как умножение матриц, входные данные и веса должны иметь один и тот же тип с плавающей точкой. Эта функция проверяет, что все tensors имеют один тип, проверяет, что этот тип — dtype (если он указан), и возвращает его. Тип должен быть типом с плавающей точкой. Если не указан ни tensors, ни dtype, функция вернёт dtypes.float32.
| Аргументы | |
|---|---|
tensors | Тензоры входных значений. Могут содержать элементы None, которые будут проигнорированы. |
dtype | Ожидаемый тип. |
| Возвращает | |
|---|---|
| Проверенный тип. |
| Вызывает исключение | |
|---|---|
ValueError | если не указан ни tensors, ни dtype, либо результат не является типом с плавающей точкой, либо общий тип входных данных не является типом с плавающей точкой. |
© 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
