tensorflow - Tensorflow 2.0 での勾配計算

Question

Tensorflow 2.0 からの私の例を次に示します。

import tensorflow as tf

w = tf.Variable([[1.0]])
with tf.GradientTape() as tape_1:
    loss_1 = w * w


with tf.GradientTape() as tape_2:
    loss_2 = w * w * w

grad_1 = tape_1.gradient(loss_1, w)
grad_2 = tape_2.gradient(loss_2, w)
print(grad_1)
print(grad_2)

戻ります:

tf.Tensor([[2.]], shape=(1, 1), dtype=float32)
tf.Tensor([[3.]], shape=(1, 1), dtype=float32)

上記は正しい係数ですが、grad_23w^2 であることも示しているはずです。部分を取得するにはどうすればよいw^2ですか?