ynd/Theano — comment on commit cf80a8ef84e896ef88156f8478c6191518cc880e
File:  line 0
By nouiz:

I forgot, can you implement the Dot.infer_shape method? You can probably copy the implementation of the tensor.Dot class. Just do a small test that when your function return the shape of your Dot, the dot get removed by optimization and that you have the good shape.