用tensorflow实现正向传播和反向传播的一个小demo

用例:输入采用numpy生成一个10×5的矩阵作为X,用numpy生成一个1×5的矩阵作为系数W

函数:y = W·X(点乘)

代码:

# coding:utf-8

import tensorflow as tf
import numpy as np


a = np.random.random((10, 5))
matrix1 = tf.convert_to_tensor(a)
c = np.random.random((1, 5))
matrix3 = tf.convert_to_tensor(c)
print('输入矩阵:\n', a)
print('点乘系数:\n', c)


def f(x, w):
    product_1 = tf.multiply(x, w)  # 矩阵点乘(内积)
    return product_1


y = f(matrix1, matrix3)
# print(y)
# exit()

dy = tf.gradients(y, matrix1)
with tf.Session() as sess:
    print('点乘函数运行完毕,输出为:\n', sess.run(y))
    print('求导(反向传播)求出点乘系数:\n', sess.run(dy))

使用tensorflow的求导方法,输出结果和预期一致,

即:正向-》知道输入矩阵、系数(或称为函数、算法)可以求出输出矩阵,

       反向-》知道输入矩阵、输出矩阵,可以通过求导方法求出系数(或称为函数、算法)

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值