x = tf.Variable(tf.zeros((2, 2))) ➊
with tf.GradientTape() as tape:
    y = 2 * x + 3
grad_of_y_wrt_x = tape.gradient(y, x) ➋

➊ 크기가 (2, 2)고 초깃값이 모두 0인 변수를 생성합니다.

➋ grad_of_y_wrt_x는 (x와 크기가 같은) (2, 2) 크기의 텐서로 x = [[0, 0], [0, 0]]일 때 y = 2 * x + 3의 곡률을 나타냅니다.

W = tf.Variable(tf.random.uniform((2, 2)))
b = tf.Variable(tf.zeros((2,)))
x = tf.random.uniform((2, 2)) 
with tf.GradientTape() as tape:
    y = tf.matmul(x, W) + b ➊  
grad_of_y_wrt_W_and_b = tape.gradient(y, [W, b]) ➋

➊ matmul은 텐서플로의 점곱 함수입니다.

➋ grad_of_y_wrt_W_and_b는 2개의 텐서를 담은 리스트입니다. 각 텐서는 W, b와 크기가 같습니다.