提交 81b32a53 authored 作者: abalkin's avatar abalkin 提交者: Frederic

Added Eig.grad stub

上级 8d7a8d56
...@@ -12,7 +12,8 @@ from theano.tensor.opt import (register_stabilize, ...@@ -12,7 +12,8 @@ from theano.tensor.opt import (register_stabilize,
register_specialize, register_canonicalize) register_specialize, register_canonicalize)
from theano.gof import local_optimizer from theano.gof import local_optimizer
from theano.gof.opt import Optimizer from theano.gof.opt import Optimizer
from theano.gradient import grad_not_implemented
1
try: try:
import scipy.linalg import scipy.linalg
imported_scipy = True imported_scipy = True
...@@ -918,4 +919,12 @@ class Eig(Op): ...@@ -918,4 +919,12 @@ class Eig(Op):
def __str__(self): def __str__(self):
return "Eig" return "Eig"
def grad(self, (x,), g_outputs):
"""The gradient function should return :math:`dw, dv`, where
.. math:: dw = ...
.. math:: dv = ...
"""
return [grad_not_implemented(self, 0, x, "Work in progress.")]
eig = Eig() eig = Eig()
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论