theano_scan_demo_compute_Jacobian_matrix

最新推荐文章于 2017-02-08 20:41:42 发布

hebastast

最新推荐文章于 2017-02-08 20:41:42 发布

阅读量328

点赞数 1

分类专栏：机器学习数据挖掘

本文链接：https://blog.csdn.net/HE19930303/article/details/52745395

版权

机器学习同时被 2 个专栏收录

19 篇文章 0 订阅

订阅专栏

数据挖掘

14 篇文章 0 订阅

订阅专栏

import numpy as np
import theano.tensor as T
import theano
floatX='float32'

V=T.vector('V')
A=T.matrix('A')
y=T.tanh(T.dot(V,A))

results,updates=theano.scan(lambda i:T.grad(y[i],V),sequences=[T.arange(y.shape[0])])
compute_jac_v=theano.function([V,A],results,allow_input_downcast=True)

x=np.eye(5,dtype=theano.config.floatX)[0]
w=np.eye(5,3,dtype=theano.config.floatX)
w[2]=np.ones((3),dtype=theano.config.floatX)

print(compute_jac_v(x,w))

[[ 0.41997434  0.          0.41997434  0.          0.        ]
 [ 0.          1.          1.          0.          0.        ]
 [ 0.          0.          1.          0.          0.        ]]