W3cubDocs

/TensorFlow 2.9

tf.experimental.numpy.experimental_enable_numpy_behavior

Enable NumPy behavior on Tensors.

Enabling NumPy behavior has three effects:

  • It adds to tf.Tensor some common NumPy methods such as T, reshape and ravel.
  • It changes dtype promotion in tf.Tensor operators to be compatible with NumPy. For example, tf.ones([], tf.int32) + tf.ones([], tf.float32) used to throw a "dtype incompatible" error, but after this it will return a float64 tensor (obeying NumPy's promotion rules).
  • It enhances tf.Tensor's indexing capability to be on par with NumPy's.
Args
prefer_float32 Controls whether dtype inference will use float32 for Python floats, or float64 (the default and the NumPy-compatible behavior).

© 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