Hands-On Machine Learning with TensorFlow.js

Visualizing an operation graph

Operation graphs are at the core of modern machine learning frameworks, including TensorFlow. They are powerful and flexible constructs that allow us to build any kind of mathematical structure technically. Let's take a look at this important structure in more detail.

The following diagram is an example of an operation graph that represents a simple classification neural network. As you can see, the operation graph illustrates the flow of the data. The squared nodes represent the tensors for the input, output, and intermediate nodes. Each tensor transformation is represented by oval nodes. These are responsible for changing the shape of the tensor, the values inside it, and their control flow. This indicates ...