tf.experimental.numpy.experimental_enable_numpy_behavior
Включить поведение NumPy для тензоров.
tf.experimental.numpy.experimental_enable_numpy_behavior(
prefer_float32=False
)
Включение поведения NumPy имеет три следствия:
- Это добавляет к
tf.Tensorнекоторые общие методы NumPy, такие какT,reshapeиravel. - Это изменяет продвижение типа данных в операторах
tf.Tensorдля соответствия NumPy. Например,tf.ones([], tf.int32) + tf.ones([], tf.float32)раньше вызывал ошибку «несовместимый тип данных», но после этого он будет возвращать тензор float64 (соблюдая правила продвижения NumPy). - Это расширяет возможности индексирования
tf.Tensorдля соответствия NumPy.
| Аргументы | |
|---|---|
prefer_float32 | Управляет тем, будет ли вывод типа данных использовать float32 для чисел с плавающей точкой Python или float64 (по умолчанию и поведение, совместимое с NumPy). |
© 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/versions/r2.9/api_docs/python/tf/experimental/numpy/experimental_enable_numpy_behavior