Open
Description
TensorFlow implements a subset of the NumPy API, available as
tf.experimental.numpy
. This allows running NumPy code, accelerated by TensorFlow, while also allowing access to all of TensorFlow's APIs. It will be available in the stable branch starting from TensorFlow 2.4.
Several benefits of using TF Numpy
- More complete NDArray API.
- Reduce memory copy, especially in distributed training scenarios.
- Significantly speed up your NumPy operations using Tensorflow and its new NumPy API