































import theano
import theano.tensor as T
# 符号变量
W = T.dmatrix('W') # 参数
V = T.dmatrix('V') # 方向向量
x = T.dvector('x') # 输入
y = T.dot(x, W) # 函数 y = x·W
# 计算 Rop: dy/dW · V
jvp = T.Rop(y, W, V)
f = theano.function([W, V, x], jvp)
# 代入数值
print(f(
[[1, 1], [1, 1]], # W
[[2, 2], [2, 2]], # V
[0, 1] # x
)) # 输出: [2. 2.]

本博客是博主个人学习时的一些记录,不保证是为原创,个别文章加入了转载的源地址,还有个别文章是汇总网上多份资料所成,在这之中也必有疏漏未加标注处,如有侵权请与博主联系。 如果未特殊标注则为原创,遵循 CC 4.0 BY-SA 版权协议。
此内容由惯性聚合(RSS阅读器)自动聚合整理,仅供阅读参考。 原文来自 — 版权归原作者所有。