Skip to content

Instantly share code, notes, and snippets.

@benanne
benanne / consider_constant.py
Created February 25, 2014 16:12
Theano consider constant op
import theano
import theano.tensor as T
from theano.tensor.opt import register_canonicalize
class ConsiderConstant(theano.compile.ViewOp):
def grad(self, args, g_outs):
return [T.zeros_like(g_out) for g_out in g_outs]
consider_constant = ConsiderConstant()
register_canonicalize(theano.gof.OpRemove(consider_constant), name='remove_consider_constant')