x = torch.tensor(3.0, requires_grad=True)
y = x ** 2        # node: power
z = 2 * y + 1     # nodes: multiply, then add