tf.experimental.numpy.experimental_enable_numpy_behavior
Stay organized with collections
Save and categorize content based on your preferences.
Enable NumPy behavior on Tensors.
tf.experimental.numpy.experimental_enable_numpy_behavior(
prefer_float32=False
)
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).
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2023-10-06 UTC.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2023-10-06 UTC."],[],[]]