fixing diagdot apply node
@@ -4,6 +4,8 @@ from __future__ import division, print_function
__all__ = ["DiagDotOp"]
+import theano
+from theano import gof
import theano.tensor as tt
from .base_op import CeleriteBaseOp
@@ -16,6 +18,11 @@ class DiagDotOp(CeleriteBaseOp):
num_input = 2
output_ndim = (1, )
+ def make_node(self, *args):
+ in_args = [tt.as_tensor_variable(a) for a in args]
+ out_args = [tt.vector(dtype=theano.config.floatX).type()]
+ return gof.Apply(self, in_args, out_args)
+
def grad(self, inputs, gradients):
A, B = inputs
return (
