20 - 多输出感知机梯度

7个月前 247次点击 来自 TensorFlow

收录专题: TensorFlow入门笔记

x=tf.random.normal([2,4])
w=tf.ones([4,3])
b=tf.ones([3])
y = tf.constant([2,0])


with tf.GradientTape() as tape:
    tape.watch([w, b])
    prob=tf.nn.softmax(x@w+b,axis=1)
    loss=tf.reduce_mean(tf.losses.MSE(tf.one_hot(y,depth=3),prob))

grads = tape.gradient(loss, [w, b])
print('w grad:', grads[0])
print('b grad:', grads[1])
Card image cap
开发者雷

尘世间一个小小的开发者,每天增加一些无聊的知识,就不会无聊了

要加油~~~

技术文档 >> 系列应用 >>
热推应用
Let'sLearnSwift
学习Swift的入门教程
PyPie
Python is as good as Pie
标签