一睹为快,自动梯度求导

作者: 焜俞 | 来源:发表于2020-03-19 15:08 被阅读0次

    一睹为快,自动梯度求导

    笔者不想过多的谈TF有什么用,只是提供一个入门TF的教程

    本节目录

    • 自动梯度求导

    自动梯度求导

    自动梯度求导可以说是所有的深度学习框架所必须的工具,无论是TF,Torch等。我们首先就来介绍一下TF的自动梯度求导机制:

    import tensorflow as tf
    
    x = tf.Variable(10.0,dtype = tf.float32)#声明变量x,并赋值为10,类型是float31
    
    with tf.GradientTape() as tape: #自动求导记录
        y = x**2 + x + 1 #记录y和x关系
    gradient = tape.gradient(y,x) #进行自动梯度求导
    print(gradient) #打印结果
    
    

    运行结果如下

    tf.Tensor(21.0, shape=(), dtype=float32)
    

    相关文章

      网友评论

        本文标题:一睹为快,自动梯度求导

        本文链接:https://www.haomeiwen.com/subject/jkklyhtx.html