#NEWLINE# def infer_shape( self , node , ishapes ) : #NEWLINE# #INDENT# if ( len ( ishapes ) == 3 ) : #NEWLINE# #INDENT# ( xshp , ind1shp , ind2shp ) = ishapes #NEWLINE# if ( ( len ( xshp ) == 2 ) and ( ind1shp is not None ) and ( len ( ind1shp ) == 1 ) and ( ind2shp is not None ) and ( len ( ind2shp ) == 1 ) ) : #NEWLINE# #INDENT# if ( node . inputs [ 2 ] . owner is None ) : #NEWLINE# #INDENT# return [ ind2shp ] #NEWLINE# #UNINDENT# else : #NEWLINE# #INDENT# return [ ind1shp ] #NEWLINE# #UNINDENT# #UNINDENT# #UNINDENT# raise theano . tensor . basic . ShapeError ( 'case not implemented' )